继续畅通工程（克鲁斯卡尔算法变形）

Problem Link：点击打开链接

题目描述

    省政府“畅通工程”的目标是使全省任何两个村庄间都可以实现公路交通（但不一定有直接的公路相连，只要能间接通过公路可达即可）。现得到城镇道路统计表，表中列出了任意两城镇间修建道路的费用，以及该道路是否已经修通的状态。现请你编写程序，计算出全省畅通需要的最低成本。 

输入描述:

    测试输入包含若干测试用例。每个测试用例的第1行给出村庄数目N ( 1< N < 100 )；随后的 N(N-1)/2 行对应村庄间道路的成本及修建状态，每行给4个正整数，分别是两个村庄的编号（从1编号到N），此两村庄间道路的成本，以及修建状态：1表示已建，0表示未建。

    当N为0时输入结束。

输出描述:

    每个测试用例的输出占一行，输出全省畅通需要的最低成本。

输入例子:

3
1 2 1 0
1 3 2 0
2 3 4 0
3
1 2 1 0
1 3 2 0
2 3 4 1
3
1 2 1 0
1 3 2 1
2 3 4 1
0

输出例子:

3
1
0

AC code：

#include<iostream>
#include<algorithm>
#include<stdio.h>
#include<map>
#include<math.h>
#include<string.h>
#include<queue>
#include<vector>
#include<set>
#define LL long long
#define exp 1e-9
#define MAXN 1000010        
using namespace std;

struct Road{
	int u,v;
	int cost;
}road[5010];

bool cmp(Road a,Road b)
{
	return a.cost<b.cost;
}

int N,M,sum,cnt;
int fa[111];

void init()
{
	int i;
	for(i=1;i<=N;i++)
	{
		fa[i]=i;
	}
}

int fin(int r)
{
	int tmp,son;
	son=r;
	while(r!=fa[r])
		r=fa[r];
	while(r!=tmp)
	{
		tmp=fa[son];
		fa[son]=r;
		son=tmp;
	}
	return r;
}

bool join(int r1,int r2)
{
	int x,y;
	x=fin(r1);
	y=fin(r2);
	if(x!=y)
	{
		if(x>y)
		{
			fa[x]=y;
		}
		else
		{
			fa[y]=x;
		}
		return true;
	}
	return false;
}

void Kruscal()
{
	int i;
	sum=0;
	sort(road+1,road+cnt+1,cmp);
	for(i=1;i<=cnt;i++)
	{
		if(join(road[i].u,road[i].v))
		{
			sum+=road[i].cost;
		}
	}
}

int main( )  
{  
//	freopen("D:\\in.txt","r",stdin); 
	int i,u,v,cost,state;
	while(scanf("%d",&N)!=EOF)
    {
    	if(N==0) break;
    	M=N*(N-1)/2;
    	init();
    	cnt=0;
    	for(i=1;i<=M;i++)
    	{
    		scanf("%d%d%d%d",&u,&v,&cost,&state);
    		if(state==1)
    			join(u,v);
    		else
    		{
    			road[++cnt].u=u;
    			road[cnt].v=v;
    			road[cnt].cost=cost;
			}
		}
		Kruscal();
		printf("%d\n",sum);
	} 	 
	return 0;  
}