Hdu 5834 Magic boy Bi Luo with his excited tree(从树上每个点

传送门:Hdu 5834 Magic boy Bi Luo with his excited tree

题意:给你一棵树,有n个点,每个点都有一个利润,每条边都有一个花费,问从每个点出发,获得的最大的利润为多少(每个点上的利润只能取一次,每条边每走一次就要花费)

思路:我们任意选取一个根,每个点都保存四个值,down[i][0],down[i][1]分别表示向下不回到这个点和回到这个点所能获取的最大利润
up[i][0]和up[i][1]分别表示向上回到这个点和不回到这个点所能获取的利润的最大值

down[i][1]是能容易计算的,down[i][0]可以通过这样去计算,它是由若干条回到i这个点的和一条不会来的路径组成
先把这个点u的子节点v中down[v][1]-e(u,v)*2大于等于0的直接加上来,并把这些点进行标记,那么我们再找寻一条不回来的路径就可以了

定义up[i]不包括i这个点
up[i][1]也是很容易计算,up[i][0]的计算也可以分为两类
1.父亲结点向上返回+父亲结点向下不返回
2.父亲结点向上不返回+父亲结点向下返回

#include<bits/stdc++.h>
using namespace std;
const int maxn=1e5+7;
struct Edge{
    int to,next,cost;
}e[maxn*2];
int head[maxn],tot;
int down[maxn][2],ans[maxn],up[maxn][2],vis[maxn],val[maxn];

void init(){
    tot=0;
    memset(head,-1,sizeof(head));
    memset(vis,0,sizeof(vis));
}

void addedge(int from,int to,int cost){
    e[tot]=(Edge){to,head[from],cost};
    head[from]=tot++;
}

void dfs1(int u,int pre){
    down[u][0]=down[u][1]=val[u];
    int maxv=0,another_maxv=0;
    for(int i=head[u];i!=-1;i=e[i].next){
        int v=e[i].to;
        if(v==pre)  continue;
        dfs1(v,u);
        if(down[v][1]-2*e[i].cost>=0)
            maxv+=down[v][1]-2*e[i].cost,down[u][1]+=down[v][1]-2*e[i].cost;
    }
    for(int i=head[u];i!=-1;i=e[i].next){
        int v=e[i].to;
        if(v==pre)  continue;
        if(down[v][1]-2*e[i].cost>=0){
            if(another_maxv<maxv+down[v][0]-e[i].cost-(down[v][1]-2*e[i].cost))
                another_maxv=maxv+down[v][0]-e[i].cost-(down[v][1]-2*e[i].cost),vis[u]=v;
        }
        else
            if(another_maxv<maxv+down[v][0]-e[i].cost)
                another_maxv=maxv+down[v][0]-e[i].cost,vis[u]=v;
    }
    down[u][0]+=another_maxv;
}

void dfs2(int u,int pre){
    for(int i=head[u];i!=-1;i=e[i].next){
        int v=e[i].to;
        if(v==pre)  continue;
        //计算向上返回
        int tmp=up[u][1]+down[u][1];
        if(down[v][1]-2*e[i].cost>=0)
            up[v][1]=max(0,tmp-2*e[i].cost- (down[v][1]-2*e[i].cost) );
        else
            up[v][1]=max(0,tmp-2*e[i].cost);
        //计算向上不返回
        //第一种情况,u向上不返回+u向下返回
        tmp=up[u][0]+down[u][1];
        if(down[v][1]-2*e[i].cost>=0)
            tmp-=(down[v][1]-2*e[i].cost);
        tmp=max(tmp-e[i].cost,0);
        //第二种情况,向上返回+向下不返回
        int tmp1=up[u][1];
        if(vis[u]==v){  //重新计算
            int maxv=0,another_maxv=0;
            for(int j=head[u];j!=-1;j=e[j].next){
                int V=e[j].to;
                if(V==pre||V==v)
                    continue;
                if(down[V][1]-2*e[j].cost>=0)
                    maxv+=down[V][1]-2*e[j].cost;
            }
            for(int j=head[u];j!=-1;j=e[j].next){
                int V=e[j].to;
                if(V==pre||V==v)    continue;
                if(down[V][1]-2*e[j].cost>=0)
                    another_maxv=max(another_maxv,maxv+down[V][0]-e[j].cost-(down[V][1]-2*e[j].cost));
                else
                    another_maxv=max(another_maxv,maxv+down[V][0]-e[j].cost);
            }
            tmp1+=val[u]+another_maxv;
        }
        else{
            tmp1+=down[u][0];
            if(down[v][1]-2*e[i].cost>=0)
                tmp1-=(down[v][1]-2*e[i].cost);
        }
        tmp1=max(tmp1-e[i].cost,0);
        up[v][0]=max(tmp1,tmp);
        dfs2(v,u);
    }
}

int main(){
    int _,n,u,v,w;
    scanf("%d",&_);
    for(int case1=1;case1<=_;case1++){
        init();
        scanf("%d",&n);
        for(int i=1;i<=n;i++)
            scanf("%d",&val[i]);
        for(int i=1;i<n;i++)
            scanf("%d%d%d",&u,&v,&w),addedge(u,w),addedge(v,w);
        up[1][0]=up[1][1]=0;
        dfs1(1,0);
        dfs2(1,0);
        printf("Case #%d:\n",case1);
        for(int i=1;i<=n;i++)
            printf("%d\n",max(up[i][0]+down[i][1],up[i][1]+down[i][0]));
    }
    return 0;
}