2019 杭电多校8 Acesrc and Travel （树形dp 换根）

题目：http://acm.hdu.edu.cn/showproblem.php?pid=6662

题意：

有两个人在树上博弈，每个点节点有两个分数a[i]和b[i]，先手先选择一个点，后手在先手选的点的相邻点中选择一个点，然后先手在后手选的点的相邻点中选择一个两个人都没有走过的点，直到不能走，游戏就结束。一个人走到节点x，那么先手会获得分数a[x]，后手就会会获得分数b[x]。最后询问先手能获得与后手的差值最大值。

解法：换根+树形dp ；

代码：

#include<bits/stdc++.h>
using namespace std;
const int M =100010;
typedef long long ll;
int T,n;
ll a[M],b[M],x,y,ans,dp[M][2],pre[M][2],last[M][2],Tmp[M][2],tmp[2],t,inf=1ll<<62;
vector<int> vc[M];
void dfs(int u,int fa){
    dp[u][0]=dp[u][1]=inf;
    for(auto v:vc[u]){
        if(v!=fa){
            dfs(v,u);
            dp[u][0]=min(dp[u][0],a[u]-b[u]-dp[v][1]);
            dp[u][1]=min(dp[u][1],b[u]-a[u]-dp[v][0]);
        }
    }
    if(dp[u][0]==inf)dp[u][0]=a[u]-b[u],dp[u][1]=b[u]-a[u];
}
void Dfs(int u,int fa){
    t=inf;
    for(auto v:vc[u])t=min(t,a[u]-b[u]-dp[v][1]);
    ans=max(t,ans);
    last[u][1]=0;last[u][0]=0;
    Tmp[u][1]=inf;Tmp[u][0]=inf;
    for(auto w:vc[u]){
        if(last[u][0]==0||Tmp[u][0]>a[u]-b[u]-dp[w][1])Tmp[u][0]=a[u]-b[u]-dp[w][1],last[u][0]=w;
        if(last[u][1]==0||Tmp[u][1]>b[u]-a[u]-dp[w][0])Tmp[u][1]=b[u]-a[u]-dp[w][0],last[u][1]=w;
    }
    for(auto v:vc[u]){
        if(v!=fa){
            pre[u][0]=dp[u][0];pre[u][1]=dp[u][1];
            tmp[0]=inf;tmp[1]=inf;
            if(last[u][0]==v||last[u][1]==v){
                for(auto w:vc[u]){
                    if(w!=v){
                        tmp[0]=min(tmp[0],a[u]-b[u]-dp[w][1]);
                        tmp[1]=min(tmp[1],b[u]-a[u]-dp[w][0]);
                    }
                }
                if(tmp[0]==inf)tmp[0]=a[u]-b[u],tmp[1]=b[u]-a[u];
            }else tmp[1]=Tmp[u][1],tmp[0]=Tmp[u][0];
            dp[u][0]=tmp[0],dp[u][1]=tmp[1];
            Dfs(v,u);
            dp[u][0]=pre[u][0],dp[u][1]=pre[u][1];
        }
    }
}
int main(){
    //freopen("f.in","r",stdin);
    scanf("%d",&T);
    while(T--){
        scanf("%d",&n);
        for(int i=1;i<=n;++i)vc[i].clear();
        for(int i=1;i<=n;++i)scanf("%lld",a+i);
        for(int i=1;i<=n;++i)scanf("%lld",b+i);
        for(int i=1;i<n;++i){
            scanf("%lld%lld",&x,&y);
            vc[x].push_back(y);
            vc[y].push_back(x);
        }
        ans=1ll<<63;
        dfs(1,0);
        Dfs(1,0);
        printf("%lld\n",ans);
    }
}