洛谷P1111 :修复公路(并查集)

https://www.luogu.org/problemnew/show/P1111

题目背景

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

题目描述

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

输入输出格式

输入格式：

 

第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

解题思路：

按照修复时间从小到大排序，把点加入集合，直到把所有的点加入集合，最后加入的那个时间就是最早通车时间。

#include <stdio.h>
#include <algorithm>
using namespace std;
#define N 100010
struct data
{
	int u;
	int v;
	int w;
}e[N];
int f[1020];
int cmp(data a, data b)
{
	return a.w<b.w;
}
int getf(int v)
{
	if(f[v]==v)
		return v;
	else
	{
		f[v]=getf(f[v]);
		return f[v];
	}
}
int merge(int u, int v)
{
	int t1, t2;
	t1=getf(u);
	t2=getf(v);
	if(t1!=t2)
	{
		f[t2]=t1;
		return 1;
	}
	return 0;
}
int main()
{
	int n, m, i, count;
	scanf("%d%d", &n, &m);
	for(i=0; i<m; i++)
		scanf("%d%d%d", &e[i].u, &e[i].v, &e[i].w);
	sort(e, e+m, cmp);
	for(i=0; i<=n; i++)
		f[i]=i;
	count=0;
	for(i=0; i<m; i++)
	{
		if(merge(e[i].u, e[i].v))
			count++;
		if(count==n-1)
			break;
	}
	if(count==n-1) 
		printf("%d\n", e[i].w);
	else
		printf("-1\n");
	return 0;
}