Python CCF 201412-4 最优灌溉

【问题描述】　　
雷雷承包了很多片麦田，为了灌溉这些麦田，雷雷在第一个麦田挖了一口很深的水井，所有的麦田都从这口井来引水灌溉。
　　为了灌溉，雷雷需要建立一些水渠，以连接水井和麦田，雷雷也可以利用部分麦田作为“中转站”，利用水渠连接不同的麦田，这样只要一片麦田能被灌溉，则与其连接的麦田也能被灌溉。
　　现在雷雷知道哪些麦田之间可以建设水渠和建设每个水渠所需要的费用（注意不是所有麦田之间都可以建立水渠）。请问灌溉所有麦田最少需要多少费用来修建水渠。
　　【输入形式】　　
　　输入的第一行包含两个正整数n, m，分别表示麦田的片数和雷雷可以建立的水渠的数量。麦田使用1, 2, 3, ……依次标号。
　　接下来m行，每行包含三个整数ai, bi, ci，表示第ai片麦田与第bi片麦田之间可以建立一条水渠，所需要的费用为ci。
　　【输出形式】　　
　　输出一行，包含一个整数，表示灌溉所有麦田所需要的最小费用。
　　【样例输入】　
　　4 4
　　1 2 1
　　2 3 4
　　2 4 2
　　3 4 3
　　【样例输出】　　
　　 6
　　【样例说明】　　
　　建立以下三条水渠：麦田1与麦田2、麦田2与麦田4、麦田4与麦田3。 【评分标准】　　
　　评测用例规模与约定
　　前20%的评测用例满足：n≤5。
　　前40%的评测用例满足：n≤20。
　　前60%的评测用例满足：n≤100。
　　所有评测用例都满足：1≤n≤1000，1≤m≤100,000，1≤ci≤10,000。

def prim(graph):  #最小生成树
    n = len(graph) #n就是田的数量
    cost = [99999 for _ in range(n)]  # 父结点到该结点的边权值
    cost[0] = 0      #第一块田有井水
    visited = [False for _ in range(n)]
    t = []       #不断往t中添加田
    while len(t) < n:    # 在costs找最短边，把该最短边的结点加入t，标记为已访问
        minCost = 99999
        minNode = None
        for i in range(n):
            if not visited[i] and cost[i] < minCost:
                minCost = cost[i]
                minNode = i
        t.append(minNode)
        visited[minNode] = True
        for edge in graph[minNode]:# 遍历该结点的边，更新最短边
            if not visited[edge[0]] and edge[1] < cost[edge[0]]:
                cost[edge[0]] = edge[1]
    return cost

tian, qv = map(int, input().split())# 构建带权邻接表
graph = [[] for _ in range(tian)]
for i in range(qv):
    qi, zhong, quan = map(int, input().split())
    graph[qi - 1].append([zhong - 1, quan])
    graph[zhong - 1].append([qi - 1, quan])
cost = prim(graph)
total = 0
for c in cost:
    total += c
print(total)