POJ2377-Bad Cowtractors

来源：互联网发布：dede免费源码站编辑：程序博客网时间：2024/06/04 21:16

最小生成树的变形题，最大生成树。

这道题显然用Kruskal算法可以直接计算最小生成树的值，但是题目需要判断最小生成树是否存在，因此我们引入变量node_num表示树中的节点数。

比较树中的节点树与总节点树即可。

ps.为什么是单向边？

#include <cstdio>#include <algorithm>using namespace std;const int maxn = 1000 + 10;const int maxm = 2*20000 + 10;int par[maxn];int rk[maxn];struct edge {    int from;    int to;    int cost;};edge es[maxm];int node_num;void init(int n) {    for (int i = 0; i < n; i++) {        par[i] = i;        rk[i] = 0;    }}int find(int x) {    if (par[x] == x) {        return x;    } else {        return par[x] = find(par[x]);    }}void unite(int x, int y) {    x = find(x);    y = find(y);    if (x == y) {        return;    }    if (rk[x] < rk[y]) {        par[x] = y;    } else {        par[y] = x;        if (rk[x] == rk[y]) {            rk[x]++;        }    }}bool same(int x, int y) {    return find(x) == find(y);}bool comp(const edge& e1, const edge& e2) {    return e1.cost > e2.cost;}int kruskal(int n, int m) {    sort(es, es + m, comp);    init(n);    int res = 0;    node_num = 1;    for (int i = 0; i < m; i++) {        edge e = es[i];        if (!same(e.from, e.to)) {            unite(e.from, e.to);            res += e.cost;            node_num++;        }    }    return res;}int main(int argc, char const *argv[]) {    int n, m;    scanf("%d%d", &n, &m);    for (int i = 0; i < m; i++) {        scanf("%d%d%d", &es[i].from, &es[i].to, &es[i].cost);        es[i].from--;        es[i].to--;    }    int ans = kruskal(n, m);    if (node_num < n) {        puts("-1");    } else {        printf("%d\n", ans);    }    return 0;}

0 0