AcWing 1072 树的最长路径 题解 （动态规划—DP—树形DP）

原题传送门

#include<bits/stdc++.h>

using namespace std;

const int N = 1e4 + 10;

int e[N], h[N], w[N], ne[N], idx;
int n, ans;

void add(int a, int b, int c){
	e[idx] = b;
	w[idx] = c;
	ne[idx] = h[a];
	h[a] = idx ++ ;
}

int dfs(int u, int father){//搜索经过以u为根节点，根节点的父节点为father的子树的距离的最长值 
	int dist = 0;
	int d1 = 0, d2 = 0;
	for(int i = h[u]; i != -1; i = ne[i]){
		int j = e[i];
		if(j == father) continue;
		int d = dfs(j, u) + w[i];
		dist = max(d, dist);
		
		/*
		关于判断为什么一定是d而不是dist
		dist表示以u为根节点子树的最大路径长度
		d1表示的是最长距离，d2表示的是次长距离
		d表示这条路径的距离 
		当d所代表的路径是最大值时，两者相等，不予讨论
		当d < dist时
		可能出现的情况是dist = d1
		因为dist的值是一直保持不变的
		所以可假设下列情况
		第一次循环时d = 7,dist值为7且将d1值更新为7
		第二次循环d = 6,dist依然为7，此时dist = d1，就会将dist = 7赋给d2，但是实际上只存在一个为7的路径，就会导致结果偏大
		所以应该用一直更新的d来更新d1和d2 
		*/
		if(d >= d1){  
			d2 = d1, d1 = d;
		}
		else if(d > d2) d2 = d;
	}
	ans = max(ans, d1 + d2);
	
	return dist;
}

int main()
{
	cin>>n;
	memset(h, -1, sizeof h);
	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, -1);
	
	cout<<ans<<endl;
	
	return 0;
}