畅通工程之局部最小花费问题 （并查集）

写出来的第一个并查集（kruskal）因为这里有已有的边，所以很自然地想到用并查集来做。但是这里判断是否达到条件需要查看并查集里面所有元素的个数，所以设置了一个num数组。在union的过程中更新。num[root]就是kruskal算法的集合里的节点总数啦~~~（和原始kruskal有点出入，原来的kruskal是最小生成树，只有n-1个边，所以记录选取的边的个数就好了）
还有一种方法是用prim,把已有的边的cost当做0

#include <bits/stdc++.h>
using namespace std;
//kruskal 设置一个量记录元素个数 
struct edge{
	int u,v;
	int cost;
	edge(int u,int v,int c):u(u),v(v),cost(c){
	} 
};int num[101]; vector<edge>e;
bool cmp(edge a,edge b){
	return a.cost<b.cost;
}bool visit[101];
int father[1001];
int findfather(int x){
	int a=x;
	while(x!=father[x])
	x=father [x];
	return x;
}
void unionn(int a,int b) {
	int fa=findfather(a);
	int fb=findfather(b);
	if(fa!=fb){father[fa]=fb;num[fb]+=num[fa];
	}
	
}int ans=0;
int main(int argc, char** argv) {
	fill(num,num+101,1);
	int n,m,a,b,s,co;
	cin>>n;int k=n*(n-1)/2;
	for(int i=0;i<n;i++)father[i]=i;
	for(int i=0;i<k;i++){
		scanf("%d %d %d %d",&a,&b,&co,&s);
		if(s==1){
			unionn(a,b);
		} 
		else //待选边 
		e.push_back(edge(a,b,co));
	}
	
	sort(e.begin(),e.end(),cmp);
	for(int i=0;i<e.size();i++)
	{
		int fa=findfather(e[i].u);
		int fb=findfather(e[i].v);
		if(fa!=fb){
			father[fa]=fb;
			num[fb]+=num[fa];
			ans+=e[i].cost;
			if(num[fb]==n)break; 
		}
		
	}
	cout<<ans;
		return 0;
	}