hdu 2196 Computer (换根dp模板)

题目链接：点击这里

题目大意：
求一棵树中所有节点能达到的最长距离

题目分析：
一个节点能到达的最长距离有两种情况，第一种是到其子树中的点中的最大的距离（记为正向最长路），另一种是其父亲节点的不经过该点的最长距离+该点到其父亲节点的距离（记为反向最长路）
设 $dp[u][0]$ 和 $dp[u][1]$ 分别表示 $u$ 到 $u$ 的子树中的最大距离和次大距离，即 $u$ 的正向最长路和正向次长路， $dp[u][2]$ 表示其父亲节点的不经过该点的最长距离+该点到其父亲节点的距离，即反向最长路
$dp[u][0]$ 和 $dp[u][1]$ 可以通过简单的树形 $dp$ 处理出来（代码中的$dfs1$）
记 $u$ 的父亲节点为 $fa$
$dp[u][2]$ 的求解过程有两种可能：
第一种是 $u$ 不是 $fa$ 正向最长路上的点，此时 $dp[u][2]=max(dp[fa][0],dp[fa][2])+dis(u,v)$
第二种是 $u$ 是 $fa$ 正向最长路上的点，此时 $dp[u][2]=max(dp[fa][1],dp[fa][2])+dis(u,v)$
此过程可通过一次自上而下的dfs递推出来（代码中的$dfs2$）

具体细节见代码：

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
#include<cmath>
#include<vector>
#define ll long long
#define inf 0x3f3f3f3f
using namespace std;
int read()
{
	int res = 0,flag = 1;
	char ch = getchar();
	while(ch<'0' || ch>'9')
	{
		if(ch == '-') flag = -1;
		ch = getchar();
	}
	while(ch>='0' && ch<='9')
	{
		res = (res<<3)+(res<<1)+(ch^48);//res*10+ch-'0';
		ch = getchar();
	}
	return res*flag;
}
const int maxn = 1e5+5;
const int mod = 1e9+7;
const double pi = acos(-1);
const double eps = 1e-8;
vector<pair<int,int>> v[maxn];
int dp[maxn][3],son[maxn]; //son[u]表示u的正向最长路要经过的儿子
void dfs1(int u,int fa) //处理dp[u][0],dp[u][1]
{
	dp[u][0] = dp[u][1] = dp[u][2] = 0;
	for(int i = 0;i < v[u].size();i++)
	{
		int to = v[u][i].first,dis = v[u][i].second;
		if(to == fa) continue;
		dfs1(to,u);
		if(dp[u][0] < dp[to][0]+dis) //如果发现更优的最大值则更新 
		{
			dp[u][1] = max(dp[u][1],dp[u][0]);
			dp[u][0] = dp[to][0]+dis;
			son[u] = to; //u的正向最大距离要经过儿子to 
		}
		else dp[u][1] = max(dp[u][1],dp[to][0]+dis); //否则更新次大值 
	}
}
void dfs2(int u,int fa) //处理得到dp[u][2] 
{
	for(int i = 0;i < v[u].size();i++)
	{
		int to = v[u][i].first,dis = v[u][i].second;
		if(to == fa) continue;
		if(son[u] == to) dp[to][2] = max(dp[u][1],dp[u][2])+dis;
		else dp[to][2] = max(dp[u][0],dp[u][2])+dis;
		dfs2(to,u);
	}
}
int main()
{
	int n;
	while(~scanf("%d",&n))
	{
		for(int i = 1;i <= n;i++)
			v[i].clear();
		for(int i = 2;i <= n;i++)
		{
			int to = read(),dis = read();
			v[i].push_back(make_pair(to,dis));
			v[to].push_back(make_pair(i,dis));
		}
		dfs1(1,1);
		dfs2(1,1);
		for(int i = 1;i <= n;i++)
			printf("%d\n",max(dp[i][0],dp[i][2]));
	}
	return 0;
}

2021/8/21 update代码风格：

//#pragma GCC optimize(2)
//#pragma GCC optimize("Ofast","inline","-ffast-math")
//#pragma GCC target("avx,sse2,sse3,sse4,mmx")
#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
#include<cmath>
#include<vector>
#include<set>
#include<map>
#include<stack>
#include<queue>
#define ll long long
#define inf 0x3f3f3f3f
#define int  ll
#define endl '\n'
#define IOS ios::sync_with_stdio(0); cin.tie(0); cout.tie(0)
using namespace std;
int read()
{
	int res = 0,flag = 1;
	char ch = getchar();
	while(ch<'0' || ch>'9')
	{
		if(ch == '-') flag = -1;
		ch = getchar();
	}
	while(ch>='0' && ch<='9')
	{
		res = (res<<3)+(res<<1)+(ch^48);//res*10+ch-'0';
		ch = getchar();
	}
	return res*flag;
}
const int maxn = 1e4+5;
const int mod = 998244353;
const double pi = acos(-1);
const double eps = 1e-8;
int n,son[maxn],dp[maxn][3];
vector<pair<int,int>>g[maxn];
void dfs1(int u,int fa)
{
	dp[u][0] = dp[u][1] = dp[u][2] = son[u] = 0;
	for(auto p:g[u])
	{
		int to = p.first,val = p.second;
		if(to == fa) continue;
		dfs1(to,u);
		if(dp[u][0] < dp[to][0]+val)
		{
			dp[u][1] = dp[u][0];
			dp[u][0] = dp[to][0]+val;
			son[u] = to;
		}
		else dp[u][1] = max(dp[u][1],dp[to][0]+val);
	}
}
void dfs2(int u,int fa)
{
	for(auto p:g[u])
	{
		int to = p.first,val = p.second;
		if(to == fa) continue;
		if(son[u] == to) dp[to][2] = max(dp[u][1],dp[u][2])+val;
		else dp[to][2] = max(dp[u][0],dp[u][2])+val;
		dfs2(to,u); 
	}
}
signed main()
{
	while(~scanf("%d",&n))
	{
		for(int i = 1;i <= n;i++)
			g[i].clear();
		for(int i = 2;i <= n;i++)
		{
			int to = read(),dis = read();
			g[i].push_back(make_pair(to,dis));
			g[to].push_back(make_pair(i,dis));
		}
		dfs1(1,1);
		dfs2(1,1);
		for(int i = 1;i <= n;i++) printf("%lld\n",max(dp[i][0],dp[i][2]));
	}
	return 0;
}