1304: [CQOI2009]叶子的染色

题目链接

题目大意：m个结点的无根树，可以选择一个度数大于1的结点作为根，然后给一些结点着黑/白色，着色需要保证根结点到每个叶子的简单路径上都至少包含一个有色结点，给出根到每个叶子路径上最后一个有色节点的颜色，最小化着色点数

题解：如果确定了根，只需要dp一下就可以求出答案了
f[i][0/1]表示以i为根的子树，i为0/1的代价

可以随便选一个非叶子节点为根……
Orz

我的收获：……

#include <iostream>
#include <cstdio>
#include <cstring>
#include <algorithm>
using namespace std;

const int M=10005;
#define INF 0x3f3f3f3f

int t,head[M];
int n,lev;
int col[M];
int f[M][2];

struct edge{int to,nex;}e[M<<1];

void add(int u,int v){e[t].to=v,e[t].nex=head[u],head[u]=t++;}

void dfs(int x,int fa)
{
    f[x][0]=f[x][1]=1;
    if(x<=lev) f[x][!col[x]]=INF;
    for(int i=head[x];i!=-1;i=e[i].nex){
        int v=e[i].to;
        if(v==fa) continue;
        dfs(v,x);
        f[x][0]+=min(f[v][0]-1,f[v][1]);
        f[x][1]+=min(f[v][1]-1,f[v][0]);
    }
}

void work()
{
    dfs(n,0);
    printf("%d\n",min(f[n][0],f[n][1]));
}

void init()
{
    cin>>n>>lev;
    memset(head,-1,sizeof(head));
    for(int i=1;i<=lev;i++) scanf("%d",&col[i]);
    for(int x,y,i=1;i<n;i++) scanf("%d%d",&x,&y),add(x,y),add(y,x);
}

int main()
{
    init();
    work();
    return 0;
}