UVALive 6619 LIKE vs CANDLE(树形DP)

12 篇文章 1 订阅
思路:

              树形dp。因为最终答案要求的是差值,所以对于每个节点我们只需要考虑差值即可,dp[u][0]表示u节点以下(含u)中like的val比candle的val大的最大值,dp[u][1]表示candle的val值比like大的最大值。所以,对于每个节点我们就可以对于每个子树v:

            dp[u][0] += dp[v][0];

            dp[u][1] += dp[v][1];  

 但是对于原来已经翻转过的点,我们要无偿交换swap(dp[u][0], dp[u][1]);这点应该很明确(因为我们本来求的是没有翻转过的,现在由于这一点本来翻转过,导致我们本来求的东西都是相反的,所以要交换一下)。

这样的话就可以使得两个值都最大,然后再考虑翻转的情况,如果翻转的话,那么可以这样更新dp值。

            dp[u][0] = max(dp[u][0], dp[u][1] - sub);

            dp[u][1] = max(dp[u][1], dp[u][0] - sub);

sub表示翻转需要的代价。

            然后需要注意的一点就是对于树根0,我们不能翻转,根据题目可以很容易看出来。


#include<iostream>
#include<cstdio>
#include<cstring>
#include<cmath>
#include<vector>
using namespace std;
const int maxn = 50000+500;
vector<int>e[maxn];
int w[maxn];
int vis[maxn];
int dp[maxn][2];
int n,x,y;
void dfs(int u)
{
    dp[u][0]=w[u];
	dp[u][1]=-w[u];
	for(int i = 0;i<e[u].size();i++)
	{
		int v = e[u][i];
		dfs(v);
		dp[u][0]+=dp[v][0];
		dp[u][1]+=dp[v][1];
	}
	if(u!=0)
	{
		if(vis[u])
			swap(dp[u][0],dp[u][1]);
		dp[u][0]=max(dp[u][0],dp[u][1]-(vis[u]?y:x));
		dp[u][1]=max(dp[u][1],dp[u][0]-(vis[u]?y:x));
	}
}
int main()
{
    while(scanf("%d%d%d",&n,&x,&y)!=EOF)
	{
		for(int i = 0;i<=n;i++)
			e[i].clear();
		memset(dp,0,sizeof(dp));
		memset(w,0,sizeof(w));
		memset(vis,0,sizeof(vis));
		for(int i = 1;i<=n;i++)
		{
			int fa,flag;
            scanf("%d%d%d%d",&w[i],&fa,&vis[i],&flag);
			e[fa].push_back(i);
			if(flag)
				w[i]=-w[i];
		}
		dfs(0);
		if(dp[0][0]<0)
			printf("HAHAHAOMG\n");
		else
			printf("%d\n",dp[0][0]);
	}
}


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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值