CCF CSP 2014年12月第4题 最优灌溉（并查集+Kruskal算法）

问题描述

试题编号：	201412-4
试题名称：	最优灌溉
时间限制：	1.0s
内存限制：	256.0MB
问题描述：	问题描述 　　雷雷承包了很多片麦田，为了灌溉这些麦田，雷雷在第一个麦田挖了一口很深的水井，所有的麦田都从这口井来引水灌溉。 　　为了灌溉，雷雷需要建立一些水渠，以连接水井和麦田，雷雷也可以利用部分麦田作为“中转站”，利用水渠连接不同的麦田，这样只要一片麦田能被灌溉，则与其连接的麦田也能被灌溉。 　　现在雷雷知道哪些麦田之间可以建设水渠和建设每个水渠所需要的费用（注意不是所有麦田之间都可以建立水渠）。请问灌溉所有麦田最少需要多少费用来修建水渠。 输入格式 　　输入的第一行包含两个正整数n, m，分别表示麦田的片数和雷雷可以建立的水渠的数量。麦田使用1, 2, 3, ……依次标号。 　　接下来m行，每行包含三个整数a i, b i, c i，表示第a i片麦田与第b i片麦田之间可以建立一条水渠，所需要的费用为c i。 输出格式 　　输出一行，包含一个整数，表示灌溉所有麦田所需要的最小费用。 样例输入 4 4 1 2 1 2 3 4 2 4 2 3 4 3 样例输出 6 样例说明 　　建立以下三条水渠：麦田1与麦田2、麦田2与麦田4、麦田4与麦田3。 评测用例规模与约定 　　前20%的评测用例满足：n≤5。 　　前40%的评测用例满足：n≤20。 　　前60%的评测用例满足：n≤100。 　　所有评测用例都满足：1≤n≤1000，1≤m≤100,000，1≤c i≤10,000。

解题思路：一道裸的最小生成树，直接套Kruskal算法就行了。

代码如下：

#include <cstdio>
#include <algorithm>

#define MAXN 1005
#define MAXM 1000005
#define LLONG long long

using std::sort;

typedef struct {
    int from,to;
    LLONG cost;
} Edge;

Edge edges[MAXM];
int par[MAXN],rank[MAXN];
int n,m;

void init() {
    for(int i = 0; i <= n; i++) {
        par[i] = i;
        rank[i] = 0;
    }
}

int find(int x) {
    return x == par[x] ? x : par[x] = find(par[x]);
}

void unite(int x,int y) {
    x = find(x);
    y = find(y);
    if(x == y) return ;
    if(rank[x] < rank[y]) {
        par[x] = y;
    } else {
        par[y] = x;
        if(rank[x] == rank[y]) {
            rank[x]++;
        }
    }
}

int same(int x, int y) {
    return find(x) == find(y);
}

bool cmp(Edge a, Edge b) {
    return a.cost < b.cost;
}

LLONG Kruskal() {
    sort(edges, edges + m,cmp);
    LLONG res = 0;
    for(int i = 0; i < m; i++) {
        Edge e = edges[i];
        if(same(e.from, e.to)) continue;
        unite(e.from, e.to);
        res += e.cost;
    }
    return res;
}

int main(void) {
    while(scanf("%d %d",&n, &m) != EOF) {
        init();
		for(int i = 0; i < m; i++) {
            scanf("%d %d %lld",&edges[i].from, &edges[i].to, &edges[i].cost);
        }
        printf("%lld\n", Kruskal());
    }
    return 0;
}