bzoj1304 [CQOI2009]叶子的染色 dfs+树形dp

这个类型的题应该是做过很多次，但做起来还是比较慢

这个题常规方法是枚举根，但这样是n^2的，这样就考虑优化

可以作为根的点一定构成一棵树，所以每次找相邻的点一定可以考虑所有情况

然后就是如何用父节点的最优值更新子节点的最优值

首先对于一个根节点，当它的选择确定了之后，所有子树就是独立的

所以只用枚举选择（选择只缺0还是1）

然后转移到子节点就直接把这棵子树对父节点的贡献减掉就可以了

码：

#include<iostream>
#include<cstdio>
#include<vector>
using namespace std;
#define N 100005
vector<int>v[N];
int f[N][3],ff[N][3],n,m,i,j,a[N],x,y,ans=1000000009,lin0,lin1;
void dp(int o,int fa)
{
	int i,nd;
	if(v[o].size()==1)
	{
		f[o][a[o]]=0;
		f[o][a[o]^1]=1000000009;
		return ;
	}
	for(i=0;i<v[o].size();i++)
	{
	nd=v[o][i];if(nd==fa)continue;	
    dp(nd,o);
	f[o][0]+=min(f[nd][0],f[nd][1]+1);
	f[o][1]+=min(f[nd][1],f[nd][0]+1);
	}	
}
void chuli(int o,int fa)
{
	//这一个要把1都填了 
	lin0=ff[fa][0]-min(f[o][0],f[o][1]+1);//不包含他这个子树的 父亲为根 
	lin1=ff[fa][1]-min(f[o][1],f[o][0]+1);
	ff[o][0]=f[o][0]+min(lin0,lin1+1);
	//这一个要把0都填了 
	ff[o][1]=f[o][1]+min(lin0+1,lin1);
	ans=min(ans,ff[o][0]+1);
	ans=min(ans,ff[o][1]+1);
}
void dfs(int o,int fa)
{
	if(fa)chuli(o,fa);
	int i,nd;
	for(i=0;i<v[o].size();i++)
	{
		nd=v[o][i];
		if(nd==fa||v[nd].size()==1)continue;
		dfs(nd,o);		
	}	
}
int main()
{
	scanf("%d%d",&m,&n);
	for(i=1;i<=n;i++)scanf("%d",&a[i]);
	for(i=1;i<m;i++)
	{
		scanf("%d%d",&x,&y);
	   v[x].push_back(y);
	   v[y].push_back(x);
	}
	for(i=1;i<=m;i++)
	{
		if(v[i].size()>1)break;		
	}
	dp(i,0);
	ff[i][0]=f[i][0];
	ff[i][1]=f[i][1];
	ans=min(ff[i][0],ff[i][1])+1;
	dfs(i,0);	
	printf("%d",ans);
}