Cyaron&Python 生成一个无向图

#!/usr/bin/env python

from cyaron import *
# 引入CYaRon的库

# array to list (int)
_n = ati([0, 7, 50, 1E4]) # ati函数将数组中的每一个元素转换为整形，方便您可以使用1E4一类的数来表示数据大小
_m = ati([0, 11, 100, 1E4])

# 这是一个图论题的数据生成器，该题目在洛谷的题号为P1339
for i in range(1, 4):
    # 即在[1, 4)范围内循环，也就是从1到3
    test_data = IO(file_prefix="heat", data_id=i)
    # 生成 heat[1|2|3].in/out 三组测试数据

    n = _n[i] # 点数
    m = _m[i] # 边数
    s = randint(1, n) # 源点，随机选取一个
    t = randint(1, n) # 汇点，随机选取一个
    test_data.input_writeln(n, m, s, t) # 写入到输入文件里，自动以空格分割并换行

    graph = Graph.graph(n, m, weight_limit=5) # 生成一个n点，m边的随机图，边权限制为5
    test_data.input_writeln(graph) # 自动写入到输入文件里，默认以一行一组u v w的形式输出
    
    # test_data.output_gen("D:\\PyProject\\main.exe")
    # 标程编译后的可执行文件，不需要freopen等，CYaRon自动给该程序输入并获得输出作为.out