本题链接:点击打开链接
本题大意:
输入n行数据。每行数据前两个表示该条路连通的两个村庄的编号,第三个表示修该条路的成本。最后的0或1表示该路未修或已修过,求在此道路基础上要使道路畅通的最小成本。
解题思路:
本题基本上也是使用的kruskal算法,仅仅是多加了一个道路状态,无论该路已修或未修,都要把它连到树上。然后就是看加不加成本的问题。
抑或採取一种方法。当道路已修。直接把成本改为0。然后依照常规kruskal算法写就可以;本代码未用此法,详细请參见代码:
#include<stdio.h>
#include<algorithm>
using namespace std;
int per[110];
int n;
struct node{
int u,v,w,p;
}a[5000];
int cmp(node a,node b)
{
return a.w<b.w;
}
void init()
{
for(int i=1;i<=n;i++)
per[i]=i;
}
int find(int x)
{
if(x==per[x])
return x;
else
return per[x]=find(per[x]);
}
int join(node a[],int m)
{
int sum=0;
for(int i=0;i<m;i++)
{
int fx=find(a[i].u);
int fy=find(a[i].v);
if(fx!=fy)
{
per[fx]=fy;
if(!a[i].p)
sum+=a[i].w;
}
}
return sum;
}
int main()
{
while(scanf("%d",&n)!=EOF&&n)
{
int m=n*(n-1)/2;
init();
for(int i=0;i<m;i++)
{
scanf("%d%d%d%d",&a[i].u,&a[i].v,&a[i].w,&a[i].p);
if(a[i].p)
{
int fx=find(a[i].u);
int fy=find(a[i].v);
if(fx!=fy)
per[fx]=fy;
}
}
sort(a,a+m,cmp);
printf("%d
",join(a,m));
}
return 0;
}