hdu 1879 继续畅通工程 （ Kruskal ）

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;
}