【树形dp】叶子的颜色

【题目描述】
给一棵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 有边相连。

【输出】
仅一个数，即着色结点数的最小值

【样例输入】
5 3
0
1
0
1 4
2 5
4 5
3 5
【样例输出】
2

【思路】

这道题显然 采用树形dp。
当我们在更新的时候，我们发现我们需要知道子树还缺少什么颜色。
因此我们定义三个状态：dp[u][0,1,2]。分别表示子树缺少白色，黑色或者不缺颜色的最小代价。
那么：
$$dp[u][1]=\sum min(dp[v][1],dp[v][2])$$
$$dp[u][0]=\sum min(dp[v][0],dp[v][2])$$
$$dp[u][2]=min(\sum dp[v][2],dp[u][1]+1,dp[u][0]+1)$$
代码：

#include<cstdio>
#include<iostream>
#include<queue>
#include<algorithm>
#include<cstring>
#include<string>
#include<vector>
#include<cmath>
#define re register
using namespace std;
int a,b,n,m,k;
struct node{
	int u,v;
}e[200001];
int f[200001];
int nxp[200001];
int cnt=0;
int du[10001];
int root;
inline void add(int u,int v)
{
	du[u]++;
	du[v]++;
	if(du[v]>1)root=v;
	if(du[u]>1)root=u;	
	e[++cnt].u=u;
	e[cnt].v=v;
	nxp[cnt]=f[u];
	f[u]=cnt;
	e[++cnt].u=v;
	e[cnt].v=u;
	nxp[cnt]=f[v];
	f[v]=cnt;
}
inline int red()
{
    int data=0;int w=1; char ch=0;
    ch=getchar();
    while(ch!='-' && (ch<'0' || ch>'9')) ch=getchar();
    if(ch=='-') w=-1,ch=getchar();
    while(ch>='0' && ch<='9') data=(data<<3)+(data<<1)+ch-'0',ch=getchar();
    return data*w;
}
int dp[10001][3];
int dfs(int u,int fa)
{
	int sum0=0,sum1=0;
	for(int re i=f[u];i;i=nxp[i])
	{
		int v=e[i].v;
		if(v==fa)continue;
		dfs(v,u);
		dp[u][2]+=dp[v][2];
		dp[u][0]+=min(dp[v][0],dp[v][2]);
		dp[u][1]+=min(dp[v][1],dp[v][2]);
	}
	dp[u][2]=min(dp[u][2],min(dp[u][0],dp[u][1])+1);
}
int main()
{
	m=red();n=red();
	for(int re i=1;i<=n;i++)
		scanf("%d",&a),dp[i][a]=0,dp[i][a^1]=1e6,dp[i][2]=1;
	for(int re i=1;i<m;i++)add(red(),red());
	dfs(root,0);
	printf("%d",dp[root][2]);
}