HDUOJ-1879(继续畅通工程)(最小生成树)
继续畅通工程
Time Limit: 2000/1000 MS (Java/Others) Memory Limit: 32768/32768 K (Java/Others)Total Submission(s): 17892 Accepted Submission(s): 7711
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
注意:这道题中有的道路已经修建,因此在连接村庄时应优先考虑这些道路,因为这些道路已修建,所以成本可以理解为0,其次再考虑先修建成本小的道路。剩下的部分和HDUOJ_1863相似。
My solution:
/*2015.8.11*/
#include<stdio.h>
#include<algorithm>
using namespace std;
int per[110];
struct stu
{
int u,v,w,z;
} node[11000];/*注意:最大时有100个村庄,共有大约5000多种组合,因此数组要开大些*/
void chishihua()
{
int i;
for(i=1;i<110;i++)
per[i]=i;
}
int cmp(stu a,stu b)
{
if(a.z!=b.z)
return a.z>b.z;/*在排序时优先把已修建的道路放在数组前面,已修建的道路z=1,未修建的z=0;*/
else
return a.w<b.w;/*其次再按成本从小到大的顺序排列未休建的道路*/
}
int find(int x)
{
int i,j;
i=x;
while(x!=per[x])
x=per[x];
while(i!=per[i])
{
j=per[i];
per[i]=x;
i=j;
}
return x;
}
bool join(int x,int y)
{
int fx,fy;
fx=find(x);
fy=find(y);
if(fx!=fy)
{
per[fy]=fx;
return true;
}
else
return false;
}
int main()
{
int i,j,n,m,sum;
while(scanf("%d",&n)==1&&n)
{
chishihua();
m=n*(n-1)/2;/*需要记录数据的行数*/
sum=0;
for(i=0;i<m;i++)
scanf("%d%d%d%d",&node[i].u,&node[i].v,&node[i].w,&node[i].z);
sort(node,node+m,cmp);
for(i=0;i<m;i++)
{
if(join(node[i].u,node[i].v))
{
if(node[i].z==0)
sum+=node[i].w;
}
}
printf("%d\n",sum);
}
return 0;
}