BZOJ1864 [Zjoi2006]三色二叉树 树形DP

一道树形DP入门题，不需要多叉树转二叉树。

用f(i,j)表示第i个节点涂j颜色时，其子树顶点为绿色最多的个数，fs(i,j)则表示最少的个数。很容易写出状态转移方程，用记忆化搜索实现。

然后对于输入，可以利用DFS或栈来转化成一棵树。

貌似这个代码还可以继续精简，没必要这么长。

//BZOJ1864
#include
#include
#include
#include
#include
#include
#include
#include
//Red 0 Green 1 Blue 2
using namespace std;
const int MAXN=500010;
const int INF=1e9;
struct Node
{
	int ch[2],s,f[3],fs[3];
	Node() {f[0]=f[1]=f[2]=-1;fs[0]=fs[1]=fs[2]=-1;}
}a[MAXN];
stack S[2];
char str[MAXN];
int len,ans;
int getf(int x,int c)
{
	if(a[x].f[c]!=-1) return a[x].f[c];
	if(a[x].s==0)
	{
		if(c==1) a[x].f[c]=1;
		else a[x].f[c]=0;
		return a[x].f[c];
	}
	if(a[x].s==1)
	{
		for(int i=0;i<=2;i++)
			if(i!=c) a[x].f[c]=max(a[x].f[c],getf(a[x].ch[0],i));
		if(c==1) a[x].f[c]++;
		return a[x].f[c];
	}
	if(a[x].s==2)
	{
		int l,r;
		for(int i=0;i<=2;i++) if(i!=c) {l=i;break;}
		for(int i=0;i<=2;i++) if(i!=c&&i!=l) {r=i;break;}
		a[x].f[c]=max(a[x].f[c],getf(a[x].ch[0],l)+getf(a[x].ch[1],r));
		a[x].f[c]=max(a[x].f[c],getf(a[x].ch[0],r)+getf(a[x].ch[1],l));
		if(c==1) a[x].f[c]++;
		return a[x].f[c];
	}
}
int getfs(int x,int c)
{
	if(a[x].fs[c]!=-1) return a[x].fs[c];
	if(a[x].s==0)
	{
		if(c==1) a[x].fs[c]=1;
		else a[x].fs[c]=0;
		return a[x].fs[c];
	}
	if(a[x].s==1)
	{
		a[x].fs[c]=INF;
		for(int i=0;i<=2;i++)
			if(i!=c) a[x].fs[c]=min(a[x].fs[c],getfs(a[x].ch[0],i));
		if(c==1) a[x].fs[c]++;
		return a[x].fs[c];
	}
	if(a[x].s==2)
	{
		int l,r;
		for(int i=0;i<=2;i++) if(i!=c) {l=i;break;}
		for(int i=0;i<=2;i++) if(i!=c&&i!=l) {r=i;break;}
		a[x].fs[c]=INF;
		a[x].fs[c]=min(a[x].fs[c],getfs(a[x].ch[0],l)+getfs(a[x].ch[1],r));
		a[x].fs[c]=min(a[x].fs[c],getfs(a[x].ch[0],r)+getfs(a[x].ch[1],l));
		if(c==1) a[x].fs[c]++;
		return a[x].fs[c];
	}
}
int main()
{
	scanf("%s",str+1);
	len=strlen(str+1);
	for(int i=0;i