叶子的染色—————树形dp

首先贴题

假如原来以x为根，y与x相邻，那么x与y的颜色不可能相同。既然颜色不同，那么将根从x变成y对答案显然也不会产生影响。

所以随便选一个点为根，然后树形DP。用dp[x][0/1]表示x的子树中，最后一个点想要得到一个白色/黑色的祖先的最小代价，转移十分容易

题目描述

给一棵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 有边相连。

输出格式：

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

输入输出样例

输入样例#1： 

5 3
0
1
0
1 4
2 5
4 5
3 5

输出样例#1： 

2

说明

M<=10000

N<=5021

我们根据分析得到递推式：

dp[root][0]+=min(dp[e[i].to][0]-1,dp[e[i].to][1]);

dp[root][1]+=min(dp[e[i].to][1]-1,dp[e[i].to][0]);

上代码：

#include
#include
#include
#include
using namespace std;
const int INF=0x3f3f3f3f;
const int maxn=10005;
int m,n,c[maxn],head[maxn],cnt,dp[maxn][2];
struct Edge{
	int to,next;
}e[maxn*2];
void add_edge(int u,int v){
    e[cnt].to=v;
	e[cnt].next=head[u];
	head[u]=cnt++;
}
void dfs(int root,int fa){
    dp[root][0]=dp[root][1]=1;
    if(root<=n){
        if(c[root]==0)
            dp[root][1]=INF;
        else
            dp[root][0]=INF;
    }
    for(int i=head[root];i!=-1;i=e[i].next)
        if(e[i].to!=fa){
            dfs(e[i].to,root);
            dp[root][0]+=min(dp[e[i].to][0]-1,dp[e[i].to][1]);
            dp[root][1]+=min(dp[e[i].to][1]-1,dp[e[i].to][0]);
        }
}
int main(){
    memset(head,-1,sizeof(head));
    cin>>m>>n;
	cnt=0;
    for(int i=1;i<=n;i++) cin>>c[i];
    for(int i=1;i>x>>y;
		add_edge(x,y);
		add_edge(y,x);
	}
    dfs(n+1,-1);
    cout<