CCF-201412-4 最优灌溉(最小生成树)

时间:2020-12-14 21:31:15

问题描述

  雷雷承包了很多片麦田,为了灌溉这些麦田,雷雷在第一个麦田挖了一口很深的水井,所有的麦田都从这口井来引水灌溉。
  为了灌溉,雷雷需要建立一些水渠,以连接水井和麦田,雷雷也可以利用部分麦田作为“中转站”,利用水渠连接不同的麦田,这样只要一片麦田能被灌溉,则与其连接的麦田也能被灌溉。
  现在雷雷知道哪些麦田之间可以建设水渠和建设每个水渠所需要的费用(注意不是所有麦田之间都可以建立水渠)。请问灌溉所有麦田最少需要多少费用来修建水渠。

输入格式

  输入的第一行包含两个正整数n, m,分别表示麦田的片数和雷雷可以建立的水渠的数量。麦田使用1, 2, 3, ……依次标号。
  接下来m行,每行包含三个整数ai, bi, ci,表示第ai片麦田与第bi片麦田之间可以建立一条水渠,所需要的费用为ci。

输出格式

  输出一行,包含一个整数,表示灌溉所有麦田所需要的最小费用。

样例输入

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≤ci≤10,000。

题解

这就是一个裸的最小生成树,没一点悬念..

#include <iostream>
#include <cstdio>
#include <algorithm>
#include <vector>
using namespace std;

const int maxn = 1000 + 5;
const int maxe = 100000 + 5;
const int inf = 0x3f3f3f3f;

struct Edge{
int u, v, cost;
Edge(){}
Edge(int u, int v, int c):u(u), v(v), cost(c){}
// bool operator < (const Edge& rhs){
// return cost < rhs.cost;
// }
};

int n, m;
int p[maxn];
Edge e[maxe];

int find(int x){ return x == p[x] ? x : p[x] = find(p[x]); }
int cmp(const Edge& e1, const Edge& e2){ return e1.cost < e2.cost; }

void Kruskal(){
for(int i = 0; i <= n; ++i) p[i] = i;
sort(e, e + m, cmp);

int ans = 0;
for(int i = 0; i < m; ++i){
int u = find(e[i].u), v = find(e[i].v), w = e[i].cost;
if(u != v){
p[u] = v;
ans += w;
}
}

cout << ans << endl;
}


int main(){
#ifdef LOCAL
freopen("data.in", "r", stdin);
#endif // LOCAL

cin >> n >> m;
for(int i = 0; i < m; ++i){
int u, v, w;
scanf("%d %d %d", &u, &v, &w);
e[i] = Edge(u, v, w);
}

Kruskal();
return 0;
}