UVALive 6619 LIKE vs CANDLE（树形DP）

思路：

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