【Luogu】P3155叶子的染色（树形DP）

　　题目链接

　　树形DP水题qwq。

　　设f[i][j]是以i为根的子树，染成j色，且满足内部需求的最少染色节点数。

　　设to是x的子节点，那么状态转移方程如此设计：

　　1、f[i][0]

　　这个状态表示i不染色，那显然很好办，对于每个to从f[to][1],f[to][2]和f[to][0]里选一个最小的即可。

　　转移方程$f[x][0]=\sum\limits_{to}min(f[to][1],f[to][2],f[to][0])$

　　2、f[i][1]

　　此时i染成黑色。那么对于每个to我们发现，既可以让它继续染白，也可以把本来染成黑色的to改为无色，让染成黑色的i来发挥to的作用。

　　于是$f[x][1]=\sum\limits_{to}min(f[to][1]-1,f[to][2])$

　　f[i][2]类似，不再赘述。

　　

#include<cstdio>
#include<cstdlib>
#include<algorithm>
#include<cctype>
#include<cstring>
#define maxn 50020
using namespace std;

inline long long read(){
    long long num=,f=;
    char ch=getchar();
    while(!isdigit(ch)){
        if(ch=='-')    f=-;
        ch=getchar();
    }
    while(isdigit(ch)){
        num=num*+ch-'';
        ch=getchar();
    }
    return num*f;
}

int f[maxn][];
int q[maxn];

struct Edge{
    int next,to;
}edge[maxn*];
int head[maxn],num;
inline void add(int from,int to){
    edge[++num]=(Edge){head[from],to};
    head[from]=num;
}

int root;
int m,n;
void dfs(int x,int fa){
    if(x<=n){
        f[x][q[x]+]=;
        f[x][(q[x]^)+]=f[x][]=;
        return;
    }
    int whi=,bla=;
    for(int i=head[x];i;i=edge[i].next){
        int to=edge[i].to;
        if(to==fa)    continue;
        dfs(to,x);
        f[x][]+=min(f[to][],min(f[to][],f[to][]));
        whi+=min(f[to][]-,f[to][]);
        bla+=min(f[to][]-,f[to][]);
    }
    f[x][]=whi+;
    f[x][]=bla+;
    return;
}

int main(){
    m=read(),n=read();
    for(int i=;i<=n;++i)    q[i]=read();
    for(int i=;i<m;++i){
        int from=read(),to=read();
        add(from,to);
        add(to,from);
    }
    root=n+;
    dfs(root,root);
    printf("%d",min(f[root][],min(f[root][],f[root][])));
    return ;
}

/*
10 5
1 0 1 1 0
1 6
6 2
6 3
7 6
7 4
7 10
10 9
9 8
8 5
*/