【BZOJ1304】[CQOI2009]叶子的染色（动态规划）

【BZOJ1304】[CQOI2009]叶子的染色（动态规划）

题面

BZOJ
洛谷

题解

很简单。
设\(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;
}