POJ 2395 Out of Hay 题解 Kruskal 最小生成树

POJ 2395 Out of Hay

题意

奶牛们的草要吃光了,这是非常可怕的事情。Bessie 决定走访每一家农场,看看她们的粮草情况。旅途艰难,Bessie 每走一公里就要喝一升水。给出连接 N 个农场的 M 条双向道路的长度,Bessie 想知道,自己走访完所有农场至少需要带一个多大的水壶。因为每到一个农场都可以灌满水壶,因此水壶容量只要够走最长的一条路就行。

思路

求最小生成树的所有边中,权值最大的一条。

代码

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
#include <iostream>
#include <stdio.h>
#include <vector>
#include <algorithm>
#define ONLINE_JUDGE

using namespace std;

const int MAX_N = 2000 + 4;
const int MAX_M = 10000 + 4;

int N;
int M;

struct Edge {
int u, v, cost;
};
bool comp(Edge & e1, Edge & e2) {
return e1.cost < e2.cost;
}

Edge edge[MAX_M];

int par[MAX_N];
int rank[MAX_N];

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

int find(int x) {
if (par[x] == x) {
return x;
}
return 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[y] = x;
}
else {
par[x] = y;
if (rank[x] == rank[y]) {
rank[y]++;
}
}
}

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

int kruskal() {
init(N);
int res = 0;
sort(edge, edge + M, comp);
for (int i = 0; i < M; i++) {
Edge e = edge[i];
if (!same(e.u, e.v)) {
unite(e.u, e.v);
res = max(res, e.cost);
}
}
return res;
}

int main() {
#ifndef ONLINE_JUDGE
freopen("in.txt", "r", stdin);
#endif

scanf("%d %d", &N, &M);
int u, v;
for (int i = 0; i < M; i++) {
scanf("%d %d %d", &u, &v, &edge[i].cost);
u--;
v--;
edge[i].u = u;
edge[i].v = v;
}
int res = kruskal();
printf("%d", res);

#ifndef ONLINE_JUDGE
fclose(stdin);
#endif
return 0;
}