洛谷3366-最小生成树-python-(kruskal+并查集)

最新推荐文章于 2022-12-21 15:26:36 发布

DongLUOWAN

最新推荐文章于 2022-12-21 15:26:36 发布

阅读量851

点赞数

分类专栏：算法-洛谷文章标签：图论最小生成树 python

本文链接：https://blog.csdn.net/DongLUOWAN/article/details/123497021

版权

算法-洛谷专栏收录该内容

37 篇文章 1 订阅

订阅专栏

这篇博客探讨了使用Python实现克鲁斯卡尔算法时遇到的时间效率问题，特别是对于大量数据导致的超时错误。作者分享了一段代码，通过全局变量和排序优化来尝试解决TLE（Time Limit Exceeded）问题，但仍然存在不足。文章最后指出，当无法找到合适的解决方案时，代码会返回'orz'作为结果。

摘要由CSDN通过智能技术生成

TLE3个点的代码,没办法，python循环太慢了。

#克鲁斯卡尔
global n,m
global side
global parent
def find(x):
    while parent[x]!=-1:
        x=parent[x]
    return x
def kruskal():
    ans=0
    #找到n-1条边就退出循环
    count=0
    for i in range(m):
        u,v,w=side[i][0],side[i][1],side[i][2]
        fa,fb=find(u),find(v)
        if fa!=fb:
            parent[fa]=fb
            ans+=w
            count+=1
        if count==n-1:
            break
    if count!=n-1:
        return -1
    else:
        return ans
n,m=map(int,input().split())
side=[[] for _ in range(m)]
parent=[-1 for _ in range(n+1)]
for i in range(m):
    x,y,z=map(int,input().split())
    side[i]+=[x,y,z]
side.sort(key=lambda x:x[2])
ans=kruskal()
if ans==-1:
    print("orz")
else:
    print(ans)