树形dp——DOVE爱旅游

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

#include
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]));
}