洛谷 P1111 修复公路

题目背景

A地区在地震过后，连接所有村庄的公路都造成了损坏而无法通车。政府派人修复这些公路。

题目描述

给出A地区的村庄数，公路是双向的。并告诉你每条公路的连着哪两个村庄，并告诉你什么时候能修完这条公路。问最早什么时候任意两个村庄能够通车，即最早什么时候任意两条村庄都存在至少一条修复完成的道路（可以由多条公路连成一条道路）

输入输出格式

输入格式：

第1行两个正整数N,M

下面M行，每行3个正整数x, y, t，告诉你这条公路连着x,y两个村庄，在时间t时能修复完成这条公路。

输出格式：

如果全部公路修复完毕仍然存在两个村庄无法通车，则输出-1，否则输出最早什么时候任意两个村庄能够通车。

输入输出样例

输入样例#1：

4 4
1 2 6
1 3 4
1 4 5
4 2 3

输出样例#1：

5

说明

N≤1000,M≤100000
x≤N,y≤N,t≤100000

C++题解可以见我的博客园博文：https://www.cnblogs.com/Weixu-Liu/p/10651745.html

在这个题中，我用的是python3，注意最后在判断是否构成最小生成树的时候，注意输入的点的个数是否最后递减到1，如果是，输出时间，如果不是，输出-1。

python3代码：

n, m = map(int, input().split())
father, edge, ans = [i for i in range(n + 1)], [], -1

def find(x):
    while x != father[x]:
        father[x] = find(father[x])
        x = father[x]
    return x

def merge(a, b):
    ax = find(a)
    bx = find(b)
    if ax == bx: return 0
    else:
        father[ax] = bx
        return 1
for i in range(m):
    u, v, w = map(int, input().split())
    edge.append((u, v, w))
edge.sort(key=lambda e : e[2])
for e in edge:
    if merge(e[0], e[1]) == 1:
        # ans += e[2]
        ans = e[2]
        n -= 1
if n == 1:
    print(ans)
else:
    print(-1)