Codeforces 275D Zero Tree 树形DP+贪心

点击打开链接

题意:n个结点的树,每个结点有权值v[i],操作:选以1为根的子树,使子树上的点全部+1/-1.

n<=1e5,v[i]<=1e9 问最少需要多少次操作 使得树上结点值都为0?


贪心:先使叶子结点变0最优,否则非叶子点先变0,由于之后要操作其叶子,又会变为非0
总的操作数为结点1的加减次数,设dp[u][0/1] u为根子树全为0时 u的加/减次数

例如:u的子节点v1,v2需要加3,5次 则只需要5次+操作 前三次(v1,v2),后两次(v2)
dp[u][0]=max(dp[u][0],dp[v][0]) 然后在根据a[u]的值更新dp[u]即可

#include <bits/stdc++.h>using namespace std;typedef long long ll;const int N=2e5+20;const ll mod=1e9+7;vector<int> e[N],tmp,t;ll n,a[N],deg[N],fa[N];ll sum[N];//sum[u] uµ½rootËùÔÚµÄÁ´±»²Ù×÷Á˶àÉÙ´Î ll dp[N][2];// dp[u][0] u×ÓÊ÷Ϊ0ʱ,½áµãu,+1µÄ´ÎÊý void dfs(int u,int par){dp[u][1]=dp[u][0]=0;for(int i=0;i<e[u].size();i++){int v=e[u][i];if(v==par)continue;dfs(v,u);dp[u][1]=max(dp[u][1],dp[v][1]);dp[u][0]=max(dp[u][0],dp[v][0]);}a[u]=a[u]+dp[u][0]-dp[u][1];if(a[u]>0)dp[u][1]+=a[u];elsedp[u][0]+=abs(a[u]);//cout<<u<<' '<<dp[u][0]<<' '<<dp[u][1]<<endl;}int main(){while(cin>>n){int u,v;memset(deg,0,sizeof(deg));for(int i=1;i<=n-1;i++){scanf("%d%d",&u,&v);e[u].push_back(v);e[v].push_back(u);deg[u]++,deg[v]++;}for(int i=1;i<=n;i++)scanf("%I64d",&a[i]);dfs(1,0);cout<<dp[1][0]+dp[1][1]<<endl;for(int i=1;i<=n;i++)e[i].clear();}return 0;}