最后一道畅通工程题,居然wrong了好几次,我囧!!!刚开始是数组开小了,然后是出栈,后来又是超时,好吧,改错的哥都错了!
囧!这个代码效率不是很高660ms;
这题做法延续上几题的做法,大的东西没有变!
又是并查集+快排!好吧,总归是过了,最后一题畅通工程!!!
#include"algorithm"
#include<math.h>using namespace std;
int pre[2000];
int find(int k)
{
if(k==pre[k])
return k;
pre[k]=find(pre[k]);
return pre[k];
}
struct node
{
int x,y,z,w;
}a[40000];
int cmp(node a,node b)
{
return a.z<b.z;
}
int main()
{
int i,n,f1,f2,ans,t;
while(scanf("%d",&n),n)
{
t=n*(n-1)/2;
for(i=1;i<=n;i++)
pre[i]=i;
for(i=0;i<t;i++)
scanf("%d %d %d %d",&a[i].x,&a[i].y,&a[i].z,&a[i].w);
for(i=0;i<t;i++)
{
if(a[i].w==1)
{
f1=find(a[i].x);
f2=find(a[i].y);
if(f1!=f2)
{
pre[f1]=f2;
}
}
}
sort(a,a+t,cmp);
ans=0;
for(i=0;i<t;i++)
{
if(a[i].w==0)
{
f1=find(a[i].x);
f2=find(a[i].y);
if(f1!=f2)
{
pre[f1]=f2;
ans+=a[i].z;
}
}
}
printf("%d\n",ans);
}
return 0;
}
题目链接:http://acm.hdu.edu.cn/showproblem.php?pid=1879