HDU 5834 Magic boy Bi Luo with his excited tree

树形dp。

先dfs一次处理子树上的最优解，记录一下回到这个点和不回到这个点的最优解。

然后从上到下可以推出所有答案。细节较多，很容易写错。

#pragma comment(linker, "/STACK:1024000000,1024000000")
#include<cstdio>
#include<cstring>
#include<cmath>
#include<algorithm>
#include<vector>
#include<map>
#include<set>
#include<queue>
#include<stack>
#include<iostream>
#include<ctime>
using namespace std;
typedef long long LL;
const double pi=acos(-1.0);
void File()
{
    freopen("D:\\in.txt","r",stdin);
    freopen("D:\\out.txt","w",stdout);
}

const int maxn=;
vector<int>G[maxn],t[][maxn][];
int T,n,v[maxn],dp[maxn][],ans[maxn],sz;
struct Edge{int u,v,w;}e[*maxn];
bool flag[maxn];

void add(int a,int b,int c)
{
    e[sz].u=a; e[sz].v=b; e[sz].w=c;
    G[a].push_back(sz++);
}

void dfs(int x,int f)
{
    flag[x]=; dp[x][]=dp[x][]=v[x];

    int i; if(f==) i=; else i=G[x].size()-;
    while()
    {
        if(f==&&i==G[x].size()) break; if(f==&&i==-) break;

        int id=G[x][i],to=e[id].v;
        if(flag[to])
        {
            t[f][x][].push_back(dp[x][]); t[f][x][].push_back(dp[x][]);
            if(f==) i++; else i--; continue;
        }

        dfs(to,f);

        int t1=dp[x][], t2=dp[x][]+dp[to][]-e[id].w, t3=dp[x][]+dp[to][]-*e[id].w;
        dp[x][]=max(t1,max(t2,t3));
        dp[x][]=dp[x][]+max(dp[to][]-*e[id].w,);

        t[f][x][].push_back(dp[x][]); t[f][x][].push_back(dp[x][]);
        if(f==) i++; else i--;
    }

}

void get(int x,int a,int b,int c)
{
    flag[x]=;

    int t1=dp[x][];
    int t2=dp[x][]+b-c;
    int t3=a+dp[x][]-*c;

    ans[x]=max(t1,max(t2,t3));

    int SZ=G[x].size();

    for(int i=;i<SZ;i++)
    {
        int id=G[x][i],to=e[id].v;
        if(flag[to]) continue;
        int na=,nb=,dp0=,dp1=, x1=,x2=,y1=,y2=;
        if(i->=) x1=t[][x][][i-];
        if(SZ--(i+)>=) x2=t[][x][][SZ--(i+)];
        if(i->=) y1=t[][x][][i-];
        if(SZ--(i+)>=) y2=t[][x][][SZ--(i+)];
        dp0=x1+x2; if(x1!=&&x2!=) dp0=dp0-v[x];
        int k1,k2;
        k1=x1+y2;  if(x1!=&&y2!=) k1=k1-v[x];
        k2=x2+y1;  if(x2!=&&y1!=) k2=k2-v[x];
        dp1=max(k1,k2);
        dp0=max(dp0,v[x]); dp1=max(dp1,v[x]);
        na=max(dp0,dp0-*c+a); nb=max(dp1,max(dp0+b-c,a+dp1-*c));
        get(to,na,nb,e[id].w);

    }

}

int main()
{
    scanf("%d",&T); int cas=;
    while(T--)
    {
        scanf("%d",&n); sz=;

        memset(dp,,sizeof dp);
        for(int i=;i<=n;i++)
        {
            G[i].clear();
            t[][i][].clear(); t[][i][].clear();
            t[][i][].clear(); t[][i][].clear();
        }

        for(int i=;i<=n;i++) scanf("%d",&v[i]);
        for(int i=;i<=n-;i++)
        {
            int u,v,w; scanf("%d%d%d",&u,&v,&w);
            add(u,v,w); add(v,u,w);
        }

        memset(flag,,sizeof flag); dfs(,);
        memset(flag,,sizeof flag); dfs(,);

        memset(flag,,sizeof flag); get(,,,);

        printf("Case #%d:\n",cas++);
        for(int i=;i<=n;i++) printf("%d\n",ans[i]);
    }
    return ;
}