【CQOI2009】叶子的染色[树形dp]

[CQOI2009]叶子的染色

(f[x][0/1/2])分别表示x点染为黑/白/不染x为根的子树满足条件的最小染色点数

然后就是注意初始化ovo

#include<bits/stdc++.h>
using namespace std;
#define ll long long
#define rg register
#define Max(x,y) ((x)>(y)?(x):(y))
#define Min(x,y) ((x)>(y)?(y):(x))
const int N=10000+5,M=2e5+5,inf=0x3f3f3f3f,P=19650827;
int n,m,rt,c[N],f[N][4],in[N];
template <class t>void rd(t &x){
    x=0;int w=0;char ch=0;
    while(!isdigit(ch)) w|=ch=='-',ch=getchar();
    while(isdigit(ch)) x=(x<<1)+(x<<3)+(ch^48),ch=getchar();
    x=w?-x:x;
}

int head[N],tot=0;
struct edge{int v,nxt;}e[N<<1];
void add(int u,int v){
	e[++tot]=(edge){v,head[u]},head[u]=tot;
}

void dfs(int u,int fa){
	if(u<=n) f[u][c[u]]=1,f[u][c[u]^1]=inf,f[u][2]=1;
	else f[u][0]=f[u][1]=1;
	for(int i=head[u],v;i;i=e[i].nxt){
		v=e[i].v;
		if(v==fa) continue;
		dfs(v,u);
		f[u][0]+=Min(f[v][2],Min(f[v][0]-1,f[v][1]));
		f[u][1]+=Min(f[v][2],Min(f[v][0],f[v][1]-1));
		f[u][2]+=Min(f[v][2],Min(f[v][0],f[v][1]));
	}
}

int main(){
	freopen("in2.txt","r",stdin);
	//freopen("xor.out","w",stdout);
	rd(m),rd(n);
	for(int i=1;i<=n;++i) rd(c[i]);
	for(int i=1,u,v;i<m;++i){
		rd(u),rd(v),add(u,v),add(v,u);
	}
	rt=n+1;
	dfs(rt,0);
	printf("%d",Min(f[rt][0],Min(f[rt][1],f[rt][2])));
	return 0;
}