UVALive 6619 LIKE vs CANDLE（树形DP）

来源：互联网发布：老馆艳后知乎编辑：程序博客网时间：2024/05/29 03:58

思路：

树形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");elseprintf("%d\n",dp[0][0]);}}

0 0