链接
题意
给出一个树,边含权,对树中的每个节点求出与相距最远点的距离。
思路
该题目是树形dp的一种经典模型。
树形dp的代码总是写在dfs中的,我们在递归中对每个节点返回其后代解并取最优值,但是对于这个问题,最优解不一定存在在某点的后代中,也可能是通过该点和其父节点的路径的某个解,这样就把解分成了两部分。
我们通过两次dfs解决该问题。
在第一次dfs中,对每个节点求出通过其子得到的最优解和次优解,并记录得到最优解的子节点id。
在第二次dfs中,设v为u的一个子节点,并且u已得到全局最优解,如果u的最优解不是通过v得到的,那么用u的最优解加上u-v的边权,同v通过其子得到的最优解作对比就可得到v的最优解。如果u的最优解是通过v得到的,那么则用次优解加u-v边权和v通过其子得到的最优解作对比,得到v的全局最优解。
显然第二次dfs中,根节点就是已得到全局最优解的节点,向下层递归可逐步得到所有节点的全局最优解。
PS:这种树形dp很典型,需要对树中的每个节点去求全局最优解问题,需要从子和父两个方向去考虑。因为树形dp总是在递归中的,我们通过一次dfs很难解决问题,因为范围是越来越小的。但是当第一次dfs结束后,我们就能拿到所有点的“父向最优解”,进而结合前一次的“子向最优解”得出全局最优解。
代码
#include <cstdio>
#include <vector>
#include <iostream>
using namespace std;
#define maxn (10010)
vector<int> son[maxn], cost[maxn];
int vis[maxn], fst[maxn], check[maxn], sec[maxn];
int dfs_son(int u)
{
vis[u] = 1;
fst[u] = sec[u] = 0;
check[u] = 0;
for(int i = 0, v, l; i < son[u].size(); i++) if(!vis[v = son[u][i]])
{
l = dfs_son(v) + cost[u][i];
if(l > fst[u]) { sec[u] = fst[u]; fst[u] = l; check[u] = v; }
else if(l > sec[u]) { sec[u] = l; }
}
vis[u] = 0;
return fst[u];
}
int dfs_fa(int u, int fa)
{
vis[u] = 1;
if(fa > fst[u]) { sec[u] = fst[u]; fst[u] = fa; check[u] = 0; }
else if(fa > sec[u]) { sec[u] = fa; }
for(int i = 0, v; i < son[u].size(); i++) if(!vis[v = son[u][i]])
{
if(check[u] != v) dfs_fa(v, fst[u] + cost[u][i]);
else dfs_fa(v, sec[u] + cost[u][i]);
}
vis[u] = 0;
}
void init(int n)
{
for(int i = 0; i <= n; i++)
{
son[i].clear();
cost[i].clear();
}
}
int main()
{
int n;
while(cin >> n)
{
init(n);
for(int i = 2, j, a; i <= n; i++)
{
scanf("%d%d", &j, &a);
son[j].push_back(i);
cost[j].push_back(a);
son[i].push_back(j);
cost[i].push_back(a);
}
dfs_son(1);
dfs_fa(1, 0);
for(int i = 1; i <= n; i++)
printf("%d\n", fst[i]);
}
return 0;
}