7-11 畅通工程之局部最小花费问题

#include<iostream>
#include<queue>
#include<cstring>
#include<algorithm>
using namespace std;
int cnt,last[1005];
int a[1005],k,ans,n,m;
struct edge{
	int u,v,w; 
}e[100005];
void add(int u,int v,int w){
	e[++cnt].v=v;
	e[cnt].u=u;
	e[cnt].w=w;
}
bool cmp(edge a,edge b){
	return a.w<b.w;
}

int find(int x){
	return x==a[x]?x:a[x]=find(a[x]);
}
bool kruskal() 
{
	sort(e+1,e+m+1,cmp);
	for(int i=1;i<=m;i++)
	{
		int u=e[i].u,v=e[i].v,w=e[i].w;
		if(find(u)!=find(v))
		{
			a[find(u)]=a[find(v)];
			k++;
			ans+=w;
		}
		if(k==n-1)
		return true; 
	 }
	 return false;
 } 
int main()
{
	cin>>n;
	m=(n-1)*n/2;
	for(int i=1;i<=n;i++) a[i]=i;
	for(int i=1,u,v,w,t;i<=m;i++)
	{
		cin>>u>>v>>w>>t;
		add(u,v,w);
		if(t&&find(u)!=find(v))
		{
			a[find(u)]=a[find(v)];
			k++;
		}
	}
	if(kruskal())
		cout<<ans;
	return 0;
 }