HDU 2196 Computer [树形dp]

题干

走你

题解

/**
*比较经典的树形dp
*跑两次深搜，都是从根1开始搜索，第一次深搜获得最大子树，
*第二次深搜是在第一次深搜的基础上，进行dp操作，获得最终结果
*一个结点最终的结果，可能来自于这个结点的父亲，也可能来自这个结点的儿子，而儿子在第一次深搜
*的时候就得出来了，所以应该从上往下搜出来自父亲的最大权值来决定是否再次更新
*/
#include<cstdio>
#include<vector>
#include<cstring>
#include<algorithm>
#define max(x,y) (x) > (y) ? (x) : (y)
#define min(x,y) (x) < (y) ? (x) : (y)
using namespace std;
struct Edge{
    int u;
    int val;
    Edge(int _u, int _val):u(_u),val(_val){}
};
vector<Edge>vec[10010];
int dp[10010];
int dfs(int root){
    if(dp[root] != -1)return dp[root];
    int size = vec[root].size();
    int ans = -1;
    for(int i=0; i<size; i++){
       int res = dfs(vec[root][i].u);
       if(res == -1){
           res = 0;
           dp[vec[root][i].u] = 0;
       }
       int val = vec[root][i].val;
       ans = max(ans, res + val);
    }
    return dp[root] = ans;
}
int result[10010];
void trees(int root, int length){
    int size = vec[root].size();
    result[root] = max(result[root], length);
    result[root] = max(result[root], dp[root]);
    int u, val, first = 0, second = -1, mymax;
    //找出第一大的值
    for(int i=0;i<size;i++){
        u = vec[root][i].u;
        val = dp[u] + vec[root][i].val;
        if(val > first)first = val;
    }
    bool flag = false;
    //找出第二大的值，因为第二大的值可能和第一大的值相同，所以，用flag标记出来
    for(int i=0;i<size;i++){
        u = vec[root][i].u;
        val = dp[u] + vec[root][i].val;
        if(!flag && first == val){
            flag = true;continue;
        }
        if(first != val && val > second)second =  val;
        else if(first == val && flag && val > second)second = val;
    }
    for(int i=0;i<size;i++){
       u = vec[root][i].u;
       val = vec[root][i].val;
       //检查这个最大值是不是来自这个子树
       if(first == dp[u] + vec[root][i].val){
         mymax = second;
       }else{
         mymax = first;
       }
       //深搜
       trees(u, max(mymax + val, length + val));
    }
}
int main(){
    //freopen("in.txt","r",stdin);
    int N,u,val;
    while(~scanf("%d",&N)){
        vec[0].clear();
        vec[1].clear();
        for(int i=2;i<=N;i++){
            vec[i].clear();
            scanf("%d%d",&u,&val);
            vec[u].push_back(Edge(i,val));
        }
        memset(dp, -1, sizeof(dp));
        int a,b;
        dfs(1);
        memset(result, -1, sizeof(result));

        trees(1, 0);

        for(int i=1;i<=N;i++){
           printf("%d\n",result[i]);
        }
    }
    return 0;
}