思路:
树形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]);
}
}