Problem Description
省政府“畅通工程”的目标是使全省任何两个村庄间都可以实现公路交通(但不一定有直接的公路相连,只要能间接通过公路可达即可)。现得到城镇道路统计表,表中列出了任意两城镇间修建道路的费用,以及该道路是否已经修通的状态。现请你编写程序,计算出全省畅通需要的最低成本。
Input
测试输入包含若干测试用例。每个测试用例的第1行给出村庄数目N ( 1< N < 100 );随后的 N(N-1)/2 行对应村庄间道路的成本及修建状态,每行给4个正整数,分别是两个村庄的编号(从1编号到N),此两村庄间道路的成本,以及修建状态:1表示已建,0表示未建。
当N为0时输入结束。
当N为0时输入结束。
Output
每个测试用例的输出占一行,输出全省畅通需要的最低成本。
Sample Input
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
Sample Output
3 1 0
题解:
做了几天的并查集的题了,本类型的题已经做了好几道了,只是改动很小。本题主要是先输入每组数据,判断一下两个城市之间是否已经联通,如果已经联通,就把他们并在一起,该组数据不做记录。如果未联通,就用一个结构体记录两个城市的数字,还有费用。之后对所有的数据进行费用排序,然后意义判断两个城市是否间接或直接联通,若已联通,则抛弃该组数据,若未联通,则并在一起,并把费用加起来。
#include <iostream>
#include <cstdio>
#include <cmath>
#include <algorithm>
using namespace std;
int root[10000];
struct node
{
int x,y,fit;
} s[10000];
bool cmp(node a,node b)
{
return a.fit<b.fit;
}
int look(int n)
{
if(n!=root[n])
root[n]=look(root[n]);
return root[n];
}
void _union(int a,int b)
{
a=look(a);
b=look(b);
if(a!=b) root[a]=b;
}
int main()
{
int a,b,c,d,n;
while(scanf("%d",&n)!=EOF&&n)
{
int i,k=0,sum=0;
for(i=0; i<=n; i++) root[i]=i;
for(i=0; i<n*(n-1)/2; i++)
{
scanf("%d%d%d%d",&a,&b,&c,&d);
if(d==1) _union(a,b);
else
{
s[k].x=a;
s[k].y=b;
s[k++].fit=c;
}
}
sort(s,s+k,cmp);
for(i=0; i<k; i++)
if(look(s[i].x)!=look(s[i].y))
{
_union(s[i].x,s[i].y);
sum+=s[i].fit;
}
printf("%d\n",sum);
}
return 0;
}