Codeforces1529C (树形dp)

题目链接
在这里插入图片描述
令dp[u][0/1] 表示u节点选取L或者R能收获的最大贡献。 考虑树形dp,最终把答案汇聚在根上。因此转移方程为
dp[u][0]+=max(dp[v][0]+abs(a[u][0]-a[v][0]),dp[v][1]+abs(a[u][0]-a[v][1]));
dp[u][1]+=max(dp[v][0]+abs(a[u][1]-a[v][0]),dp[v][1]+abs(a[u][1]-a[v][1]));


ll dp[100100][3];
vector<ll>G[100100];
ll a[100100][3];

void dfs(ll u,ll f)
{
    dp[u][0]=dp[u][1]=0;

    for(int i=0; i<G[u].size(); i++)
    {
        ll v=G[u][i];
        if(v==f)
            continue;
        dfs(v,u);
        dp[u][0]+=max(dp[v][0]+abs(a[u][0]-a[v][0]),dp[v][1]+abs(a[u][0]-a[v][1]));
        dp[u][1]+=max(dp[v][0]+abs(a[u][1]-a[v][0]),dp[v][1]+abs(a[u][1]-a[v][1]));
    }
    return ;
}
signed main()
{

    ll t;
    read(t);
    while(t--)
    {
        ll n;
        read(n);
        for(int i=1; i<=n; i++)
        {
            read(a[i][0]);
            read(a[i][1]);
        }
        memset(dp,0,sizeof(dp));

        for(int i=1; i<n; i++)
        {
            ll u,v;
            read(u);
            read(v);
            G[u].push_back(v);
            G[v].push_back(u);
        }
        dfs(1,1);
        printf("%lld\n",max(dp[1][1],dp[1][0]));
        for(int i=1; i<=n; i++)
        {
            G[i].clear();
        }

    }



}

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值