叶子的染色

给一棵m个结点的无根树，你可以选择一个度数大于1的结点作为根，然后给一些结点（根、内部结点和叶子均可）着以黑色或白色。你的着色方案应该保证根结点到每个叶子的简单路径上都至少包含一个有色结点（哪怕是这个叶子本身）。 对于每个叶结点u，定义c[u]为从根结点从U的简单路径上最后一个有色结点的颜色。给出每个c[u]的值，设计着色方案，使得着色结点的个数尽量少。

输入输出格式

输入格式：

 

第一行包含两个正整数m, n，其中n是叶子的个数，m是结点总数。结点编号为1，2，...，m，其中编号1，2，... ，n是叶子。以下n行每行一个0或1的整数（0表示黑色，1表示白色），依次为c[1]，c[2]，...，c[n]。以下m-1行每行两个整数a，b（1<=a < b <= m），表示结点a和b 有边相连。

 

输出格式：

 

仅一个数，即着色结点数的最小值。

 

#include<iostream>
#include<cstdio>
using namespace std;
const int maxm=1e4+7;
const int maxn=5037;
const int INF=0x7f7f7f7f;
int n,m,num;
int col[maxn],head[maxm],f[maxm][maxn];
struct Edge{
  int next,to;
}edge[maxm*2];
void add(int from,int to){
  edge[++num].next=head[from];
  edge[num].to=to;
  head[from]=num;
}
void dfs(int x,int pre){
  if(x<=n){
    f[x][col[x]]=1;f[x][col[x]^1]=INF; 
    return;
  } 
  f[x][0]=1;f[x][1]=1;
  for(int i=head[x];i;i=edge[i].next){
    int v=edge[i].to;
    if(v==pre) continue;
    dfs(v,x);
    f[x][0]+=min(f[v][0]-1,f[v][1]);
    f[x][1]+=min(f[v][0],f[v][1]-1);
  }
}
int main(){
  cin>>m>>n;
  for(int i=1;i<=n;i++) cin>>col[i];
  for(int i=1;i<m;i++){
    int u,v;cin>>u>>v;
    add(u,v);add(v,u);
  }
  dfs(n+1,0);
  cout<<min(f[n+1][0],f[n+1][1])<<endl;
  return 0;
}