HDU 1879 继续畅通工程

题目大意:

中文题。


解题思路:

MST,kruskal直接上就行了。


#include<cstdio>
#include<cstring>
#include<string>
#include<cmath>
#include<iostream>
#include<algorithm>
#include<set>
#define LL long long
#define db double
#define maxn 10000000
#define EPS 1e-15
#define max(a,b) ((a>b)?(a):(b))
#define min(a,b) ((a<b)?(a):(b))
using namespace std;

int m,n,p[110];
struct Fuck{
    int u,v,w,vis;
}fuck[11000],tmp;
bool cmp(Fuck x,Fuck y){
    if (x.vis!=y.vis) return x.vis<y.vis;
    return x.w<y.w;
}
int find(int x){
    while(x!=p[x]) x=p[x];
    return x;
}
void join(int x,int y){
    x=find(x);
    y=find(y);
    if (x!=y) p[x]=y;
}
void kruskal(){
    int ans=0;
    sort(fuck+1,fuck+m+1,cmp);
    for (int i=1;i<=m;i++){
        int x=find(fuck[i].u);
        int y=find(fuck[i].v);
        if (x!=y && !fuck[i].vis){
            ans+=fuck[i].w;
            p[x]=y;
        }
    }
    printf("%d\n",ans);
}
int main(){
    while (1) {
        scanf("%d",&n);
        if (n==0) break;
        m=n*(n-1)/2;
        for (int i=1;i<=n;i++) p[i]=i;
        for (int i=1;i<=m;i++){
            scanf("%d%d%d%d",&fuck[i].u,&fuck[i].v,&fuck[i].w,&fuck[i].vis);
            if (fuck[i].vis==1) join(fuck[i].u,fuck[i].v);
        }
        kruskal();
    }
    return 0;
}


你可能感兴趣的:(HDU 1879 继续畅通工程)