HDU 5834 Magic boy Bi Luo with his excited tree 树形dp

http://acm.split.hdu.edu.cn/showproblem.php?pid=5834

题意 : 每个节点有一个价值Vi，每走一次一条边要花费Ci，问从各个节点出发最多能收获多少价值

同computer那道题一样也是维护三个值...虽然比较奇怪.....

维护的是从每个点回来的最大值,不回来的次大值,不回来的最大值....

细节还挺多的,没看题解的时候对数据的综合处理部分很头痛...需要一定的代码能力...我还是太蒻了...

原题解部分在第二次dfs中对数据的综合处理部分变量设置得很巧妙...避免了我开始代码那种绕来绕去好多if最后完全找不到关系的情况...非常值得借鉴...

果然还是不要脑补手推图比较靠谱... ...

顺便一提...我这道题看错了三次题目...重新写了三次...语文能力简直max...最后还是看题解中的题目介绍才明白

参考题解:http://blog.csdn.net/angon823/article/details/52266807

AC代码如下

#include<cstdio>
#include<cstring>
#include<iostream>
#include<algorithm>
#include<cmath>
#include<queue>
using namespace std;
const int maxn=100010;
int n,m,T;
struct nod{
    int y;
    int v;
    int next;
}e[maxn*2]={};
int tot=0,head[maxn]={};
int f[maxn][3]={};//012 回来 不回来最大 不回来次大
int id[maxn]={};
int ma[maxn]={};
int c[maxn]={};
inline void init(int x,int y,int v){
    e[++tot].y=y;
    e[tot].v=v;
    e[tot].next=head[x];
    head[x]=tot;
}
inline void yu(){
    tot=0;
    memset(id,0,sizeof(id));
    memset(head,0,sizeof(head));
    memset(ma,0,sizeof(ma));
    memset(f,0,sizeof(f));
}
void dfs1(int x,int fa){
    int y,v;
    f[x][1]=c[x];
    f[x][2]=c[x];
    f[x][0]=c[x];
    int t1;
    for(int i=head[x];i;i=e[i].next){
        y=e[i].y;
        v=e[i].v;
        if(y!=fa){
            dfs1(y,x);
            if(f[y][0]-v*2>0){
                f[x][0]+=f[y][0]-v*2;
            }
        }
    }for(int i=head[x];i;i=e[i].next){
        y=e[i].y;
        v=e[i].v;
        if(y!=fa){
            t1=f[x][0]-max(0,f[y][0]-v*2)+max(0,f[y][1]-v);
            if(t1>f[x][1]){
                f[x][2]=f[x][1];
                f[x][1]=t1;
                id[x]=y;
            }
            else if(t1>f[x][2]){
                f[x][2]=t1;
            }
        }
    }
}
void dfs2(int x,int fa,int up1,int up2){
    int y,v;
    int t1,t2;
    ma[x]=max(up1+f[x][0],up2+f[x][1]);
    for(int i=head[x];i;i=e[i].next){
        y=e[i].y;
        v=e[i].v;
        t1=0,t2=0;//不回来 回来 
        if(y!=fa){
            if(id[x]==y){
                t1=max(0,f[x][2]-max(0,f[y][0]-v*2));
            }
            else{
                t1=max(0,f[x][1]-max(0,f[y][0]-v*2));
            }
            t2=max(0,f[x][0]-max(0,f[y][0]-v*2));
            t1=max(0,max(t1+up2,t2+up1)-v);
            t2=max(0,t2+up2-v*2);
            dfs2(y,x,t1,t2);
        }
    }
}
int main(){
    scanf("%d",&T);
    int t=0;
    while(T-->0){
        t++;
        yu();
        scanf("%d",&n);
        int x,v,y;
        for(int i=1;i<=n;i++){
            scanf("%d",&c[i]);
        }for(int i=1;i<n;i++){
            scanf("%d%d%d",&x,&y,&v);
            init(x,y,v);
            init(y,x,v);
        }dfs1(1,0);
        dfs2(1,0,0,0);
        printf("Case #%d:
",t);
        for(int i=1;i<=n;i++){
            printf("%d
",ma[i]);
        }
    }
}