mysterious

稳定切水!

[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;
}
阅读更多
版权声明:想转就转吧,反正也是人人都会的东西:-( https://blog.csdn.net/zichenzhiguang/article/details/52385797
文章标签: acm 树形dp
想对作者说点什么? 我来说一句

没有更多推荐了,返回首页

不良信息举报

[HDU 2196] Computer (树形dp)

最多只允许输入30个字

加入CSDN,享受更精准的内容推荐,与500万程序员共同成长!
关闭
关闭