http://acm.hdu.edu.cn/showproblem.php?pid=1879
思路:这题的排序就是先把已建的排在前面,未建的再按成本低的排在前面就可以了。
代码:
#include <iostream>
#include <stdio.h>
#include <string.h>
#include <algorithm>
using namespace std;
int map[105][105];
int p[105];
struct node
{
int a,b,p,d;
}a[100005];
bool cmp(node a,node b)
{
if(a.d!=b.d)return a.d>b.d;
else return a.p<b.p;
}
int find(int x)
{
if(p[x]==x)return p[x];
else return (p[x]=find(p[x]));
}
int kruskal(int n)
{
int sum=0;
for(int i=0;i<n;i++)
{
int fx=find(a[i].a);
int fy=find(a[i].b);
if(fx!=fy)
{
p[fy]=fx;
if(a[i].d==0)
{
sum+=a[i].p;
}
}
}
return sum;
}
int main()
{
int N;
while(scanf("%d",&N)!=EOF&&N)
{
int m=N*(N-1)/2;
int i;
for(i=0;i<=N;i++)
{
p[i]=i;
}
for(i=0;i<m;i++)
{
scanf("%d %d %d %d",&a[i].a,&a[i].b,&a[i].p,&a[i].d);
}
sort(a,a+m,cmp);
int k=kruskal(m);
cout<<k<<endl;
}
return 0;
}