HDU 1897 继续畅通工程 (Kruskal 并查集)

 

#include<iostream>
#include<cmath>
#include<algorithm>
#include<cstring>
#include<cstdio>
#include<map>
#include<vector>
#include<stack>
#include<queue>
#include<set>
#include<list>
#define mod 998244353
#define Max 0x3f3f3f3f
#define Min 0xc0c0c0c0
#define mst(a) memset(a,0,sizeof(a))
#define f(i,a,b) for(int i=a;i<b;i++)
using namespace std;
typedef long long ll;
const int maxn=105;
int pre[105];
struct Node{
    int p1,p2,cost,build;
}vil[6005];
int cmp(Node a,Node b){     //降低成本
    return a.cost<b.cost;
}
int Find(int x){    //并查集模板
    int r=x;
    while(pre[r]!=r){   //找到x的最终根节点
        r=pre[r];
    }
    int i=x,j;
    while(i!=r){        //路径压缩(路径上所有点都指向根节点)
        j=pre[i];       
        pre[i]=r;
        i=j;
    }
    return r;
}
void Join(int x,int y){
    int fx=Find(x),fy=Find(y);  
    if(fx!=fy){             //根节点不同，其中一个数作为另一个根节点
        pre[fy]=fx;
    }
}
int main(){
    ios::sync_with_stdio(false);
    int n;
    while(scanf("%d",&n)!=EOF){
        if(n==0){
            break;
        }
        for(int i=1;i<=n;i++){
            pre[i]=i;
        }
        int m=n*(n-1)/2;
        for(int i=0;i<m;i++){
            scanf("%d%d%d%d",&vil[i].p1,&vil[i].p2,&vil[i].cost,&vil[i].build);
        }
        sort(vil,vil+m,cmp);
        for(int i=0;i<m;i++){
            if(vil[i].build==1){    //先把修好的路连通
                Join(vil[i].p1,vil[i].p2);
            }
        }
        int ans=0;
        for(int i=0;i<m;i++){
            if(Find(vil[i].p1)!=Find(vil[i].p2) && vil[i].build==0){    //没连通又没修好的路，按最低成本连通
                Join(vil[i].p1,vil[i].p2);      //花钱连通未连通的路
                ans+=vil[i].cost;               //花费最低成本
            }
        }
        printf("%d\n",ans);
    }
    return 0;
}