【Problem Description】
省政府“畅通工程”的目标是使全省任何两个村庄间都可以实现公路交通(但不一定有直接的公路相连,只要能间接通过公路可达即可)。现得到城镇道路统计表,表中列出了任意两城镇间修建道路的费用,以及该道路是否已经修通的状态。现请你编写程序,计算出全省畅通需要的最低成本。
【Input】
测试输入包含若干测试用例。每个测试用例的第1行给出村庄数目N ( 1< N < 100 );随后的 N(N-1)/2 行对应村庄间道路的成本及修建状态,每行给4个正整数,分别是两个村庄的编号(从1编号到N),此两村庄间道路的成本,以及修建状态:1表示已建,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
【题解】
首先我们要知道这个“继续”是从哪来的:
畅通工程
这道题比上一道题多了一个“状态”的问题。因为有一些边是已经修了的,所以我们在Kruskal的时候希望优先选这些建好的边,这样就不用增加我的代价。因此我们要改的就是在前边排序的时候把修好的排在前面。
代码如下:
#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
const int N=100;
const int M=N*(N-1)/2;
int head[N+5],num,father[N+5];
int n,m;
struct edge
{
int u,v,w,t;
int next;
edge(){next=-1;}
}ed[4*M+5];
void build(int u,int v,int w,int t)
{
ed[++num].u=u;
ed[num].v=v;
ed[num].w=w;
ed[num].t=t;
ed[num].next=head[u];
head[u]=num;
}
int getfather(int a)
{
return father[a]==a?a:getfather(father[a]);
}
void unionn(int a,int b)
{
father[getfather(b)]=getfather(a);
}
bool cmp(edge a,edge b)//修好的放前边
{
return a.t==b.t?a.w<b.w:a.t>b.t;
}
int kruskal()
{
int ans=0,tot=0;
sort(ed+1,ed+1+num,cmp);
for(int i=1;i<=n;i++)
father[i]=i;
for(int i=1;i<=num;i++)
{
if(getfather(ed[i].v)==getfather(ed[i].u))continue;
unionn(ed[i].v,ed[i].u);
ans+=ed[i].t?0:ed[i].w;//修了的不需要付出代价
if(++tot==n-1)return ans;
}
}
int main()
{
while(scanf("%d",&n)&&n)
{
num=0;
memset(head,-1,sizeof(head));
int m=n*(n-1)/2;
for(int i=1;i<=m;i++)
{
int u,v,w,t;
scanf("%d%d%d%d",&u,&v,&w,&t);
build(u,v,w,t);
build(v,u,w,t);
}
printf("%d\n",kruskal());
}
return 0;
}