树形dp——DOVE爱旅游

最新推荐文章于 2023-03-23 23:15:57 发布

原创最新推荐文章于 2023-03-23 23:15:57 发布 · 319 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#树形dp

动态规划同时被 2 个专栏收录

23 篇文章

订阅专栏

图论

13 篇文章

订阅专栏

本文深入探讨了树形动态规划的基本概念与实现技巧，通过一个具体的编程问题，详细介绍了如何构建树形结构，设计递归函数进行状态转移，以及最终求解最优解的过程。文章适合有一定编程基础，希望掌握树形DP解题方法的读者。

DOVE爱旅游
本题一看就是树形dp，直接写模板就行了

#include<bits/stdc++.h>
using namespace std;

const int N=1000010;
int cnt=0,n,a[N],dp[N][2];
struct egde{int link,v;}q[N*2];
int head[N];
void put(int x,int y){q[++cnt].v=y,q[cnt].link=head[x],head[x]=cnt;}
void dfs(int s,int fa){
	for(int i=head[s];i;i=q[i].link){
		int v=q[i].v;
		if(v==fa) continue;
		dfs(v,s);
	    dp[s][0]+=dp[v][0];
	    dp[s][1]+=dp[v][1];
	}
	   if(a[s]==0)dp[s][0]++,dp[s][1]--;
	   if(a[s]==1)dp[s][1]++,dp[s][0]--;
	   if(a[s]==1){
	   dp[s][1]=max(dp[s][1],1);
       dp[s][0]=max(dp[s][0],0);
	   }
	   if(a[s]==0){
	   dp[s][0]=max(dp[s][0],1);
	   dp[s][1]=max(dp[s][1],0);
	   }
}
int main(){
	scanf("%d",&n);
	for(int i=1;i<=n;i++){	scanf("%d",&a[i]);}
	for(int i=1;i<n;i++){int u,v;scanf("%d%d",&u,&v);put(u,v),put(v,u);}
	for(int i=1;i<=n;i++)dp[i][1]=0,dp[i][0]=0;
	dfs(1,0);
	printf("%d",max(dp[1][1],dp[1][0]));
}