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;
}