HDU 1879 继续畅通工程

题解:最小生成树

#include <cstdio>
#include <algorithm>
using namespace std;
struct node{int a,b,l;}seg[10000];
int f[105],m,n,flag; char c;
int sf(int x){return f[x]==x?x:f[x]=sf(f[x]);}
bool cmp(node a,node b){return a.l<b.l;}
void scan(int &x){
    while(c=getchar(),c<'0'||c>'9');x=c-'0';
    while(c=getchar(),c>='0'&&c<='9')x=x*10+c-'0';
}
int main(){
    while(~scanf("%d",&n)){
        if(n==0)break;int ans=0;
        for(int i=0;i<=n;i++)f[i]=i;
        int t=n*(n-1)/2,cnt=n-1;
        for(int i=0;i<t;i++){
            scan(seg[i].a),scan(seg[i].b),scan(seg[i].l);
            scan(flag);if(flag){
               int q=sf(seg[i].a),w=sf(seg[i].b); 
               if(q!=w)cnt--,f[q]=w;
            }
        }
        sort(seg,seg+t,cmp); int i=-1;
        while(cnt){
            int q=sf(seg[++i].a),w=sf(seg[i].b);
            if(q!=w)cnt--,f[q]=w,ans+=seg[i].l;
        }
        printf("%d\n",ans);
    }
    return 0;
}
posted @ 2014-03-16 13:53  forever97  阅读(169)  评论(0编辑  收藏  举报