Week6--作业 -- D -- 数据中心[生成树]

题目描述

样例输入

4
5
1
1 2 3
1 3 4
1 4 5
2 3 8
3 4 2

样例输出

4

思路

综述

这道题主要考察的是最小生成树问题；
参考：图论、并查集。
这个题抽象出来是这样：
从该图找最小生成树；
最小生成树中输出最大的边；

过程

Kruskal过程见链接
该题和C题很类似
也是跑一边Kruskal，然后记录出最大的边权

总结

这道题最初读的时候并没有完全理解，是看到了学长提供的后面的解释才搞懂。
所以图论的题正如今晚（2020-4-1）学长所说的，会考一部分阅读理解；

代码

#include <iostream>
#include <algorithm>
using namespace std;
const int maxn = 1e5 + 5;
int n, m, root;
int tot = 0;
int par[maxn];
struct Edge {
	int u, v, w;
	bool operator < (Edge& t) {
		return w < t.w;
	}
}e[maxn];
Edge term;
void init() {
	for (int i = 0; i < maxn; i++) {
		par[i] = i;
	}
}
int find(int x) {
	if (par[x] == x)return x;
	else return par[x] = find(par[x]);
}
bool unit(int ax,int ay) {
	int x = find(ax);
	int y = find(ay);
	if (x == y)return false;
	par[x] = par[y];
	return true;
}
bool cmp(Edge a, Edge b) {return a.w < b.w;}
int kruskal() {
	std::sort(e, e +m,cmp);
	int total = 0;
	int num = 0;
	for (int i = 0; i < m; i++) {
		if (unit(e[i].u, e[i].v)) {
			total = e[i].w;
			num++;
			if (num == n)break;
		}
	}
	return total;
}

int main() {
	cin >> n >> m >> root;
	init();
	for (int i = 0; i < m; i++) {
		cin >> e[i].u >> e[i].v >> e[i].w;
	}
	cout << kruskal() << endl;
}