HDUOJ6662 Acesrc and Travel（树型DP）

HDUOJ6662 Acesrc and Travel（树型DP）

题目大意

给出一颗树，有每个节点有两个权值，现在有两个人，这两个人走过同一个顶点有不同的收益，这两个人先后选择下一个要走的顶点，两个人都会走使得最终自己收益减去对方收益尽可能大的走法，问最终两个人的差值会有多大

解题思路

对每个顶点的两个权值相减转换成一个权值之后，两个人的目的就变成了使得最终的收益尽可能大和尽可能小。对次可以得到转移式子
$$\begin{gathered} dp[u][0]=a[u]+min(dp[v][1]) \\ dp[u][1]=a[u]+max(dp[v][0]) \end{gathered}$$

AC代码

#include<bits/stdc++.h>
#define int long long  
using namespace std;
typedef long long LL;
const int size=1e5+5;
//int head[size],tot,nxt[size],to[size];
vector<int> G[size];
const LL inf=1e15;
LL dpdown[size][2][2];
int a[size],tp;
// 0 zhang's choose, 1 another choose
// 1 is greater
LL dp[size][2];
int d[size];

inline void dfs1(int u,int p)
{
    for(auto v:G[u])
    {
        if(v==p) continue;
        dfs1(v,u);d[u]++;
        if(dpdown[u][0][0]>=dpdown[v][1][0]+a[u]) dpdown[u][0][1]=dpdown[u][0][0],dpdown[u][0][0]=dpdown[v][1][0]+a[u];
        else if(dpdown[u][0][1]>dpdown[v][1][0]+a[u]) dpdown[u][0][1]=dpdown[v][1][0]+a[u];
        if(dpdown[u][1][0]<=dpdown[v][0][0]+a[u]) dpdown[u][1][1]=dpdown[u][1][0],dpdown[u][1][0]=dpdown[v][0][0]+a[u];
        else if(dpdown[u][1][1]<dpdown[v][0][0]+a[u]) dpdown[u][1][1]=dpdown[v][0][0]+a[u];
    } 
    if(!d[u]) for(int i=0;i<=1;i++) for(int j=0;j<=1;j++) dpdown[u][i][j]=a[u];
}
void dfs2(int u,int p)
{
    for(auto v:G[u])
    {
        if(v==p) continue;
        if(d[u]==1){dp[v][0]=dp[u][1]+a[v],dp[v][1]=dp[u][0]+a[v];}
        else
        {
            if(a[u]+dpdown[v][0][0]==dpdown[u][1][0]) dp[v][0]=a[v]+dpdown[u][1][1];
            else dp[v][0]=a[v]+dpdown[u][1][0];
            if(u!=1) dp[v][0]=max(dp[v][0],a[v]+dp[u][1]);
            if(a[u]+dpdown[v][1][0]==dpdown[u][0][0]) dp[v][1]=a[v]+dpdown[u][0][1];
            else dp[v][1]=a[v]+dpdown[u][0][0];
            if(u!=1) dp[v][1]=min(dp[v][1],a[v]+dp[u][0]);
        }
        dfs2(v,u);
    }
}        
int n;    
int32_t main()
{
    int t;
    //freopen("b.in","r",stdin);
    scanf("%lld",&t);
    while(t--)
    {
        memset(d,0,sizeof(d));
        scanf("%lld",&n);
        for(int i=1;i<=n;i++) scanf("%lld",&a[i]);
        for(int i=1;i<=n;i++) scanf("%lld",&tp),a[i]-=tp;
        for(int i=1;i<=n;i++) G[i].clear(),dpdown[i][0][0]=dpdown[i][0][1]=inf,dpdown[i][1][0]=dpdown[i][1][1]=-inf;
        int x,y;
        for(int i=1;i<n;i++)
        {
            scanf("%lld%lld",&x,&y);
            G[x].push_back(y);
            G[y].push_back(x);
        }
        dp[1][0]=dp[1][1]=a[1];
        dfs1(1,-1);
        dfs2(1,-1);
        LL ans=dpdown[1][0][0];
        for(int i=2;i<=n;i++) 
        {
            if(d[i]) ans=max(ans,min(dp[i][0],dpdown[i][0][0]));
            else ans=max(ans,dp[i][0]);
        }
        printf("%lld\n",ans);
    }
} 
/*
5
6
-1 -1 -1 -1 -1 -1
1 1 1 1 1 1
1 2 
2 3
3 4
4 5
5 6
5
3 2 0 0 4
4 4 2 1 2
1 2
1 3
4 3
1 5
*/