7-3 规划(一)(最小生成树，并查集）

小王生活在美丽的村庄，某天他励志想考上公务员，于是，开始了奋笔疾书。当天却遇到了一个难题，就是每个村庄与每个村庄都有很多条小道路，甚至两个村庄之间有三四条小道路。

    现在需要将这些道路整合起来，进行翻新，从泥泞路变成水泥路，每米造价是1w元，要使得每个村庄都联通，问最少需要几个w。

输入格式:

第一行输入两个整数n，m（2<=n<=1e3 && 1<=m<=1e5）分别表示村庄个数和道路个数

接下来的m行，输入三个整数 u，v，w（1<=u,v<=n && 1<=w<=1e6）表示u村到v村需要w米距离

输出格式:

输出一个整数，表示最小需要几个w

如果，所有的村庄连不起，则输出NO

输入样例:

在这里给出一组输入。例如：

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

输出样例:

在这里给出相应的输出。例如：

NO

#include<bits/stdc++.h>
using namespace std;
const int N=1e5+9;
int p[N],n,m,sum=0,ans=0;
struct node{
	int u,v,w;
}a[N];
bool cmp(node a,node b){
	return a.w<b.w;//寻找最短边
}
int find(int x){
	if(p[x]==x) return x;
	return	p[x]=find(p[x]);
}
void kruskal(){
	sort(a,a+m,cmp);
	for(int i=0;i<m;i++){
		int x=find(a[i].u);
		int y=find(a[i].v);
		if(x==y) continue;//边的前端和后端产生了圈，丢弃
		ans+=a[i].w;
		p[x]=y;
		sum++;//统计合并次数，n个点应合并n-1次
	}
}
int main(){
	cin>>n>>m;
	for(int i=0;i<N;i++) p[i]=i;//!!记得初始化
	for(int i=0;i<m;i++)
	cin>>a[i].u>>a[i].v>>a[i].w;
	kruskal();
	if(sum==n-1) cout<<ans;
	else cout<<"NO";
	return 0;
}