hdu1879浙大计算机研究生复试上机(2008)继续畅通工程

hdu1879浙大计算机研究生复试上机(2008)继续畅通工程 - 果冻虾仁

unclejelly 2021-08-21 原文

hdu1879浙大计算机研究生复试上机(2008)继续畅通工程

继续畅通工程

Time Limit: 2000/1000 MS (Java/Others) Memory Limit: 32768/32768 K (Java/Others)

Total Submission(s): 10079 Accepted Submission(s): 4399

Problem Description

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

Input

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

当N为0时输入结束。

Output

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

Sample Input

Sample Output

3
1
0

/*今天一天就做了两个MST的题，一个是prim算法实现的，另一个是kruskal算法实现的。

kruskal算法思路蛮清晰的，照着数据结构课本改改也就行了。。这次坑爹的是我改成使用了并查集来实现这一算法，为了提高点效率。。结果因为自己并查集的不熟练。导致WA了。我还找了半天找不出错来。。。哎呀。。*/

#include <iostream>
#include <queue>
#include <algorithm>
using namespace std;
const int MAX=55000;
struct Edge
{
	int vex1;
	int vex2;
	int weight;
};
int vset[MAX];
bool operator <(const Edge&E1,const Edge&E2)
{
	return E1.weight<E2.weight;
}
int GetParent(int i)
{
	if(vset[i]!=i)
		vset[i]=GetParent(vset[i]);
	return vset[i];
}
void kruskal(Edge E[],int n,int e)
{
	int k=1;
	int j=0;
	int sum=0;
	while(k<=e)
	{
		int p1=GetParent(E[j].vex1);
		int p2=GetParent(E[j].vex2);
		if(p1!=p2)
		{
			sum+=E[j].weight;
			k++;
			vset[p2]=p1;//就是这个操作。。总是手贱写成vest[E[j].vex2]=p1;其实不是啊 。。。
		}
		j++;
	}
	printf("%d\n",sum);
}
int main()
{
	int n;
	while(scanf("%d",&n))
	{
		if(!n)
			break;
		int t=n*(n-1)/2;
		int v1,v2,dist,bit;
		int e=n-1;
		Edge E[MAX];
		for(int i=1;i<=n;i++)
			vset[i]=i;
		for(int i=0;i<t;i++)
		{
			scanf("%d%d%d%d",&v1,&v2,&dist,&bit);
			E[i].vex1=v1;
			E[i].vex2=v2;
			E[i].weight=dist;
			if(bit)
			{
				int p1=GetParent(v1);
				int p2=GetParent(v2);
				if(p1==p2)
					continue;
				vset[p2]=p1;//同上。。这个总写成vest[v2]=p1;孩子们要敲起警钟啊！！！
				e--;
			}
		}
		sort(E,E+t);
		kruskal(E,n,e);
	}
	return 0;
}