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;
}