【BZOJ1304】[CQOI2009]叶子的染色（动态规划）
题面
题解
很简单。
设\(f[i][0/1/2]\)表示以\(i\)为根的子树中，还有颜色为\(0/1/2\)（\(2\)就是没有染色）的叶子节点的路径上没有任何一个染色的点。随便转移一下就好了。
#include<iostream>
#include<cstdio>
using namespace std;
#define MAX 10010
inline int read()
{
int x=0;bool t=false;char ch=getchar();
while((ch<'0'||ch>'9')&&ch!='-')ch=getchar();
if(ch=='-')t=true,ch=getchar();
while(ch<='9'&&ch>='0')x=x*10+ch-48,ch=getchar();
return t?-x:x;
}
int n,m,a[MAX];
struct Line{int v,next;}e[MAX<<1];
int h[MAX],cnt=1;
inline void Add(int u,int v){e[cnt]=(Line){v,h[u]};h[u]=cnt++;}
int f[MAX][3];
void dfs(int u,int ff)
{
if(u<=m)f[u][2]=1,f[u][a[u]^1]=1e9;
for(int i=h[u];i;i=e[i].next)
{
int v=e[i].v;if(v==ff)continue;
dfs(v,u);
f[u][2]+=f[v][2];
f[u][1]+=min(f[v][1],f[v][2]);
f[u][0]+=min(f[v][0],f[v][2]);
}
f[u][2]=min(f[u][2],min(f[u][1],f[u][0])+1);
}
int main()
{
n=read();m=read();
for(int i=1;i<=m;++i)a[i]=read();
for(int i=1;i<n;++i)
{
int u=read(),v=read();
Add(u,v);Add(v,u);
}
dfs(n,0);
printf("%d\n",f[n][2]);
return 0;
}