python之prim算法2

qq_37589983

于 2019-03-14 11:16:38 发布

阅读量271

点赞数

CC 4.0 BY-SA版权

分类专栏： python

本文链接：https://blog.youkuaiyun.com/qq_37589983/article/details/88549053

python 专栏收录该内容

3 篇文章

订阅专栏

本文探讨了使用图算法来解决寻找从特定节点到其他所有节点的最小路径问题。通过输入节点数量、边的数量和起始节点，算法构建了一个加权无向图，并运用迭代方式更新各节点的最短路径标记，最终输出从起始节点到其余节点的最小路径长度。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

提交依旧超时，但是比前面的都好

n = int(input())
m = int(input())
root = int(input())
list1 = [([0]*n) for i in range(n)]
for i in range(m):
temp = list(map(int, input().split()))
list1[temp[0]-1][temp[1]-1] = temp[2]
list1[temp[1]-1][temp[0]-1] = temp[2]
mark = n * [1000000]
mark[root-1] = 0
result = 0
no = []
for i in range(n):
if i+1 != root:
no.append(i+1)
while len(no) != 0:
minnum = 1000000
minindex = -1
for i in range(len(no)):
if list1[root-1][no[i]-1] > 0 and list1[root-1][no[i]-1] < mark[no[i]-1]:
mark[no[i]-1] = list1[root-1][no[i]-1]
for i in range(len(no)):
if mark[no[i]-1] < minnum and mark[no[i]-1] != 0:
minnum = mark[no[i]-1]
minindex = i
if result < minnum:
result = minnum
mark[no[minindex]-1] = 0
root = no[minindex]
no.pop(minindex)
print(result)