树形DP(1/6) 1072. 树的最长路径

• 题意：
  给定一棵树，树中包含 n 个结点（编号1~n）和 n−1 条无向边，每条边都有一个权值。
  现在请你找到树中的一条最长路径。
  换句话说，要找到一条路径，使得使得路径两端的点的距离最远。
  注意：路径中可以只包含一个点。
• 如下图所示
  
• 记录某个节点的子路径的最长和次长路径，即可完成更新，答案就是某个节点的最长路径和次长路径的和。
• 如果询问的是经过每个节点的最长路径，那么还需要考虑其父节点给它的贡献，即换根DP解决。

#include <bits/stdc++.h>

using namespace std;

int n,head[10010],nxt[20010],w[20010],to[20010],tot;

void add(int u,int v,int z) {
	to[++tot] = v,w[tot] = z,nxt[tot] = head[u],head[u] = tot;
}
//不同于求出经过任一点的最长简单路径的换根DP 这个普通树形DP即可
//对于每个点 考虑经过他的路径的可能的形态有哪几种
int dp[10010][2],ans;
void dfs(int u,int fa) {
	for(int i = head[u];i;i = nxt[i]) {
		int v = to[i];
		if(v == fa) continue;
		dfs(v,u);
		if(dp[v][0] + w[i] > dp[u][0]) {
			dp[u][1] = dp[u][0];
			dp[u][0] = dp[v][0] + w[i];
		} else if(dp[v][0] + w[i] > dp[u][1]) {
			dp[u][1] = dp[v][0] + w[i];
		}
	}
	ans = max(dp[u][0] + dp[u][1],ans);
}

int main() {
	cin >> n;
	for(int i = 1;i < n;i ++) {
		int a,b,c; cin >> a >> b >> c;
		add(a,b,c); add(b,a,c);
	}
	dfs(1,0);
	cout << ans << '\n';
	return 0;
}