【公路村村通】【PTA】【python】【kruskal】

现有村落间道路的统计数据表中，列出了有可能建设成标准公路的若干条道路的成本，求使每个村落都有公路连通所需要的最低成本。

输入格式:
输入数据包括城镇数目正整数N（≤1000）和候选道路数目M（≤3N）；随后的M行对应M条道路，每行给出3个正整数，分别是该条道路直接连通的两个城镇的编号以及该道路改建的预算成本。为简单起见，城镇从1到N编号。

输出格式:
输出村村通需要的最低成本。如果输入数据不足以保证畅通，则输出−1，表示需要建设更多公路。

输入样例:
6 15
1 2 5
1 3 3
1 4 7
1 5 4
1 6 2
2 3 4
2 4 6
2 5 2
2 6 6
3 4 6
3 5 1
3 6 1
4 5 10
4 6 8
5 6 3
输出样例:
12
代码长度限制
16 KB
时间限制
400 ms
内存限制
64 MB

# @author whatsupuh
# @time 2022/9/19

import operator

global num_city, num_road
num_city = 0
num_road = 0
global E
E = {}
father = {}
def Read():
    global num_city, num_road
    num_city, num_road = map(int,input().split(' '))
    for i in range(num_city+1):
        father[i] = -1
    for i in range(num_road):
        begin_road, end_road, cost = map(int, input().split())
        E[i] = (begin_road, end_road, cost)
    # print(E)
def resortE():
    global E
    E = dict(sorted(E.items(),key=lambda x : x[1][2]))
    new_key = 0
    new_E = {}
    for key, value in E.items():
        new_E[new_key] = value
        new_key += 1
    E = new_E

def findfather(x):
    if father[x] < 0:
        return x
    else:
        father[x] = findfather(father[x])
        return father[x]

def kruskal(nr):
    resortE()
    cost, num_edge = 0, 0
    for i in range(nr):
        father_begin = findfather(E[i][0])
        father_end = findfather(E[i][1])
        # father_end = E[i][1]
        if father_begin != father_end:
            cost += E[i][2]
            father[father_begin] = father_end
            num_edge += 1
            if num_edge == num_city - 1:
                break
        # print(father_begin, father_end)

    if num_edge != num_city - 1:
        print(-1)
    else:
        print(cost)

Read()
kruskal(num_road)

# print(findfather(2))