本题链接:点击打开链接
本题大意:
输入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\n",join(a,m));
}
return 0;
}