时间限制: | 1.0s |
内存限制: | 256.0MB |
问题描述: | 问题描述 雷雷承包了很多片麦田,为了灌溉这些麦田,雷雷在第一个麦田挖了一口很深的水井,所有的麦田都从这口井来引水灌溉。 输入格式 输入的第一行包含两个正整数n, m,分别表示麦田的片数和雷雷可以建立的水渠的数量。麦田使用1, 2, 3, ……依次标号。 输出格式 输出一行,包含一个整数,表示灌溉所有麦田所需要的最小费用。 样例输入 4 4 样例输出 6 样例说明 建立以下三条水渠:麦田1与麦田2、麦田2与麦田4、麦田4与麦田3。 评测用例规模与约定 前20%的评测用例满足:n≤5。 |
堆优化克鲁斯卡尔算法,纯的不能再纯了。。。。因为用py写耗费了大量时间了解排序函数,以后当板子用了
class node:
fro=to=dist=0
def getfx(x):
if fa[x]==x:
return fa[x]
fa[x] = getfx(fa[x])
return fa[x]
def merge(x,y):
tx = getfx(x)
ty = getfx(y)
fa[tx] = ty
while True:
try:
graph = []
n,m = map(int,input().split())
for i in range(m):
a,b,c = map(int,input().split())
graph.append([a,b,c])
#print(graph[i].dist)
graph = sorted(graph,key = lambda x:x[2])
# for i in range(m):
# print(graph[i][2])
fa = [0 for i in range(1005)]
for i in range(n+1):
fa[i]=i
#print(1)
# for i in range(m):
# print(graph[i].dist)
ans = 0
last = n
k = 0
while k<m and last>1:
u = graph[k][0]
v = graph[k][1]
if not (getfx(u)==getfx(v)):
merge(u,v)
last-=1
ans+=graph[k][2]
k+=1
print(ans)
except:
break