BZOJ1304: [CQOI2009]叶子的染色

题目：http://www.lydsy.com/JudgeOnline/problem.php?id=1304

树形dp。

可以发现其实根选在哪里都是没有问题的。

f[u][0],f[u][1],f[u][2]分别表示以u为根的子树全部满足条件，有0节点没有满足条件和有1节点没有满足条件。

然后就转移就好了。。

#include<cstring>
#include<iostream>
#include<cstdio>
#include<map>
#include<cmath>
#include<algorithm>
#define rep(i,l,r) for (int i=l;i<=r;i++)
#define down(i,l,r) for (int i=l;i>=r;i--)
#define clr(x,y) memset(x,y,sizeof(x))
#define maxn 50050
#define inf 2000000000
#define mm 1000000007
using namespace std;
struct data{int obj,pre;
}e[maxn*2];
int head[maxn],f[maxn][3],a[maxn],d[maxn],tot,n,m;
int read(){
    int x=0,f=1; char ch=getchar();
    while (!isdigit(ch)) {if (ch=='-') f=-1; ch=getchar();}
    while (isdigit(ch)) {x=x*10+ch-'0'; ch=getchar();}
    return x*f;
}
void insert(int x,int y){
    e[++tot].obj=y; e[tot].pre=head[x]; head[x]=tot; 
}
void dfs(int u,int fa){
    int tmp=0,tmp1=0,tmp2=0,s1=0,s2=0;
    if (a[u]!=-1) {
        f[u][0]=1; 
        if (a[u]==0) f[u][1]=0,f[u][2]=10000;
        if (a[u]==1) f[u][1]=10000,f[u][2]=0;
        return; 
    }
    for (int j=head[u];j;j=e[j].pre){
        int v=e[j].obj;
        if (v!=fa){
            dfs(v,u);
            s1=s1+f[v][1]; s2=s2+f[v][2]; tmp=tmp+f[v][0];
            tmp1+=min(f[v][0],f[v][1]); tmp2+=min(f[v][0],f[v][2]);
        }
    }
    f[u][0]=min(tmp,min(s1+1,s2+1));
    f[u][1]=min(tmp1,tmp2+1);
    f[u][2]=min(tmp1+1,tmp2);
}
int main(){
    n=read(); m=read();
    clr(a,-1);
    rep(i,1,m) a[i]=read();
    rep(i,1,n-1){
        int x=read(),y=read();
        d[x]++; d[y]++;
        insert(x,y); insert(y,x);
    }
    rep(i,1,n) if (d[i]>=2){
        dfs(i,0); printf("%d
",f[i][0]); break;
    }
    return 0;
}