Bad Cowtractors.(POJ-2377)

标签：acm   poj   

最大生成树，只需要变一下改变值的部分就可以了。

尝试了一下运用并查集的kruskal算法，与prim相比各有利弊吧，还要根据数据特点适当选取。

#include<cstdio>
#include<cstring>
#include<iostream>
#include<algorithm>
#include<queue>
#include<vector>
#include<map>
using namespace std;
int n,m,a,b,c,par[1005],rankk[1005];
struct edge {
    int u,v,cost;
    edge(int u=0,int v=0,int cost=0) : u(u),v(v),cost(cost) {}
};
bool cmp(const edge& e1,const edge& e2) {
    return e1.cost>e2.cost;
}
edge es[20005];
void init(int n) {
    for(int i=1;i<=n;i++) {
        par[i] = i;
        rankk[i] = 0;
    }
}
int findd(int x) {
    return par[x] == x ? x : par[x] = findd(par[x]);
}
void unite(int x,int y) {
    x = findd(x);
    y = findd(y);
    if(x==y) return ;
    if(rankk[x] < rankk[y]) {
        par[x] = y;
    } else {
        par[y] = x;
        if(rankk[x] == rankk[y]) rankk[x] ++ ;
    }
}
bool same(int x,int y) {
    return findd(x) == findd(y);
}
int kruskal() {
    sort(es,es+m,cmp);
    init(n);
    int res = 0;
    for(int i=0;i<m;i++) {
        edge e = edge(es[i].u,es[i].v,es[i].cost);
        if(!same(e.u,e.v)) {
            unite(e.u,e.v);
            res += e.cost;
        }
    }
    return res;
}
int main() {
    scanf("%d%d",&n,&m);
    for(int i=0;i<m;i++) {
        scanf("%d%d%d",&a,&b,&c);
        es[i] = edge(a,b,c);
    }
    int sum = kruskal();
    bool ok = true;
    for(int i=1;i<=n;i++) {
        for(int j=1;j<=n;j++) {
            if(i!=j) if(!same(i,j)) {
                ok = false; break;
            }
        }
        if(!ok) break;
    }
    if(ok) printf("%d\n",sum);
    else printf("-1\n");
    return 0;
}

Bad Cowtractors.(POJ-2377)

标签：acm   poj   

原文地址：http://blog.csdn.net/weizhuwyzc000/article/details/45417955