树的中心

树的中心

  对于顶点无赋权的图，图的中心就是找到一个点，使得该点到图中其他结点的最远距离最近。

#include<iostream>
#include<queue>
#include<algorithm>
#include<cstring>
#include<cstdio>
using namespace std;
const int MAX_N = 10010;
const int MAX_M = MAX_N * 2;
const int inf = 0x3f3f3f3f;
int n, x, y, z;
struct edge {          
	int v, w, nxt;
}e[MAX_M];
int head[MAX_N], cnt = 1;
int d1[MAX_N], d2[MAX_N];         
int up[MAX_N], f[MAX_N];
void add(int u, int v,int w) {
	e[cnt].v = v;
	e[cnt].w = w;
	e[cnt].nxt = head[u];
	head[u] = cnt++;
}
int dfs(int u, int fa) {
	d1[u] = d2[u] = -inf;
	for (int i = head[u]; ~i; i = e[i].nxt) {
		int v = e[i].v, w = e[i].w;
		if (v == fa)continue;
		int temp = dfs(v, u) + w;
		if (temp >= d1[u]) {
			d2[u] = d1[u];
			d1[u] = temp;
			f[u] = v;
		}
		else if (temp >= d2[u]) {
			d2[u] = temp;
		}
	}
	if (d1[u] == -inf)d1[u] = d2[u] = 0;
	return d1[u];
}

void dfs2(int u, int fa) {
	for (int i = head[u]; ~i; i = e[i].nxt) {
		int v = e[i].v, w = e[i].w;
		if (v == fa)continue;
		if (v == f[u]) {
			up[v] = max(up[u], d2[u]) + w;
		}
		else {
			up[v] = max(up[u], d1[u]) + w;
		}
		dfs2(v, u);
	}
}
int main() {
	cin >> n;
	memset(head, -1, sizeof head);
	cnt = 1;
	for (int i = 1; i <= n - 1; i++) {
		scanf("%d%d%d", &x, &y, &z);
		add(x, y, z);
		add(y, x, z);
	}
	dfs(1, -1);
	dfs2(1, -1);
	int ans = inf;
	for (int i = 1; i <= n; i++) {
		ans = min(ans, max(up[i], d1[i]));
	}
	cout << ans << endl;
	return 0;
}
ans = min(ans, max(up[i], d1[i]));
	}
	cout << ans << endl;
	return 0;
}