python 存储数据到有向无环图寻找路径_Python 随即生成DAG(有向无环图)

给校队选拔赛出了道DAG上的背包问题，需要生成DAG数据。

最开始使用的方法是先随机生成再判环，如果有环就重新生成。这种方法得到DAG的概率随着点数和边数的增加而急速降低，为了一个DAG要生成很多次，等很长时间。然后觉得这样的方法很stupid。。。

听了好甜给的先生成拓扑序的构造方法，这样可以保证生成的图里面没有环。

首先随机生成一个 1 到N 的permutation。这个permutation就是DAG的拓扑序，然后每次随机从前往后连边，这样就可以保证生成的是一个DAG了。真心膜拜

Life is short ,Use Python

from random import shuffle as sl

from random import randint as rd

def gn():

num = rd(1,1000)

return num

def w2f(f,num,fg):

f.write(str(num))

if fg==True:

f.write('\n')

else:

f.write(' ')

def DataMake(c):

MAXL =100000;

f = open('data'+str(c)+'.in','w')

n = 1000

node = range(1,n+1)

sl(node)

sl(node)

m = rd(1,min(n*n,5000))

w2f(f,n,0);w2f(f,m,1)

for i in range(0,m):

p1 = rd (1,n-1)

p2 = rd (p1+1,n)

x = node[p1-1]

y = node[p2-1]

l = rd(1,MAXL)

w = gn()

w2f(f,x,0);w2f(f,y,0);w2f(f,l,0);w2f(f,w,1)

k = gn()

w2f(f,k,1)

for i in range(0,k):

w2f(f,gn(),1)

print n,' node',m,' edges',k,'Queries'

f.close()

DataMake(1)

print 'Done'