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