CQOI2009 BZOJ1304 叶子的染色

传送门
先提一句，这道题确实可以像题目说的一样，任意选择根节点（原因自己想）。
这道题显然是树DP
我们用F(u,color)表示以u为节点的子树所需的最少节点数，且u刷成color。
F(i,0)表示点i涂黑色,dp(i,1)表示点i涂白色
所以有F(i,0) = sum(min(dp(i.son,1),dp(i.son,0)))当点i涂黑色时，他涂黑色的儿子可以少涂一个点，i涂白色时同理

/**************************************************************
    Problem: 1304
    User: geng4512
    Language: C++
    Result: Accepted
    Time:28 ms
    Memory:1156 kb
****************************************************************/

#include <cstdio>
#define MAXN 10005
#define min(a,b) (a)<(b)?(a):(b)
int f[MAXN][3], c[MAXN], n, m;
struct node
{
    int v;
    node *next;
}*Adj[MAXN], Edge[MAXN * 2], *Mcnt = Edge;
void addedge(int u,int v)
{
    node *t = ++Mcnt;
    t->v = v;
    t->next = Adj[u];
    Adj[u] = t;
}
void dp(int u, int fa)
{
    if(u <= n)
    {
        f[u][c[u]] = 1;
        f[u][c[u]^1] = 1<<30;
        return;
    }
    else f[u][0] = f[u][1] = 1;
    for(node *p = Adj[u]; p; p = p->next)
    {
        if(p->v == fa) continue;
        dp(p->v, u);
        f[u][1] += min(f[p->v][0], f[p->v][1] - 1);
        f[u][0] += min(f[p->v][1], f[p->v][0] - 1);
    }
    return;
}
int main()
{
    int t1, t2;
    scanf("%d%d", &m, &n);
    for(int i = 1; i <= n; i ++)
        scanf("%d", c + i);
    for(int i = 1; i <= m - 1; i ++)
    {
        scanf("%d%d", &t1, &t2);
        addedge(t1, t2);
        addedge(t2, t1);
    }
    dp(m, -1);
    printf("%d\n", min(f[m][0], f[m][1]));
    return 0;
}