您的位置:首页 > 编程语言 > Python开发

Python 随机生成DAG(有向无环图)

2013-08-21 21:55 537 查看
给校队选拔赛出了道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'
内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息
标签: