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];
	else
		dp[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;
}