mxqz
查看原帖
mxqz
526017
COsm0s楼主2023/5/30 20:25

RT.用的树的直径。

记录

#include <bits/stdc++.h>
using namespace std;
const int N = 1e6 + 10;
int n, c, d[N];
vector<int> E[N];
void dfs(int u, int fa) {
	for (int v : E[u]) {
		if (v == fa) continue;
		d[v] = d[u] + 1;
		if (d[v] > d[c]) c = v;
		dfs(v, u);
	}
}
signed main() {
	ios::sync_with_stdio(false);
	cin.tie(nullptr), cout.tie(nullptr);
	cin >> n;
	for (int i = 1; i < n; i++) {
		int u, v;
		cin >> u >> v;
		E[u].push_back(v), E[v].push_back(u);
	}
	dfs(1, 0);
	d[c] = 0, dfs(c, 0);
	cout << c;
	return 0;
}


2023/5/30 20:25
加载中...