[HDU 2196] Computer （树形dp）

链接

HDU 2196

---

题意

给出一个树，边含权，对树中的每个节点求出与相距最远点的距离。

---

思路

该题目是树形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;
}