1073. 树的中心 （树形dp）

题目
题意: 给定n个点的树，请你在树中找到一个点，使得该点到树中其他结点的最远距离最近。
思路: 树形dp。一遍dfs求出每个点到其他节点的最远距离，之后枚举一遍即可。可以将一个节点分成向下走和向上走，向下走好说，dfs呼呼走就行。如果是向上走呢?我们可以维护次长路和最长路对应哪个儿子son，对于当前点cur 的儿子j，如果他是最长路对应的儿子son，up[j] = max(up[cur],d2[cur]) + w[i];否则，up[j] = max(up[cur],d1[cur]) + w[i].因为该点向上走肯定经过父节点，又要选最长的路，分情况讨论即可。
时间复杂度: O(n)
代码:

#include<bits/stdc++.h>
using namespace std;
int n,m,k,T;
const int N = 1e5+10;
#define mem(a,x) memset(a,x,sizeof(a))
int h[N],e[N<<1],ne[N<<1],w[N<<1],idx = 0;
int d1[N]; //从i点向下走的最长路
int d2[N]; //从i点向下走的次长路
int son1[N]; //从i点向下走的最长路从哪个点下去
int up[N]; //从i点向上走的最长路
void add(int a,int b,int c)
{
	e[idx] = b,w[idx] = c,ne[idx] = h[a],h[a] = idx++;
}
int dfs(int cur,int fa) //向下的最长路
{
	for(int i=h[cur];~i;i=ne[i])
	{
		int j = e[i];
		if(j == fa) continue;
		int d = dfs(j,cur) + w[i];
		if(d >= d1[cur]) d2[cur] = d1[cur],d1[cur] = d,son1[cur] = j;
		else if(d > d2[cur]) d2[cur] = d;
	}
	return d1[cur];
}
void dfs2(int cur,int fa) //向上的最长路
{
	for(int i=h[cur];~i;i=ne[i])
	{
		int j = e[i];
		if(j == fa) continue;
		if(j == son1[cur]) //最长路
		{
			up[j] = max(d2[cur],up[cur]) + w[i];
		}
		else
		{
			up[j] = max(d1[cur],up[cur]) + w[i];
		}
		dfs2(j,cur); //还久内阁错写成dfs1
	}
}
void solve()
{
	ios::sync_with_stdio(false),cin.`tie(0),cout.tie(0);
	cin>>n; mem(h,-1); idx = 0;
	for(int i=0;i<n-1;++i)
	{
		int x,y,z; cin>>x>>y>>z;
		add(x,y,z),add(y,x,z);
	}
	dfs(1,0);
	dfs2(1,0);
	int mx = 1e9;
	for(int i=1;i<=n;++i)
	{
		// cout<<i<<":"<<d1[i]<<" "<<up[i]<<endl;
		int d = max(d1[i],up[i]);
		if(d < mx)
		{
			mx = d;
		}
	}
	cout<<mx;
}
signed main(void)
{
	solve();
	return 0;
}