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

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

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

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

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

//BZOJ1864
#include<iostream>
#include<cstdio>
#include<algorithm>
#include<string>
#include<cstring>
#include<queue>
#include<vector>
#include<stack>
//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<int> 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<str[1]-'0';i++)
	{
		S[0].push(1);
		S[1].push(i);
	}
	for(int i=2;i<=len;i++)
	{
		int p=S[0].top(),d=S[1].top();
		S[0].pop();S[1].pop();
		a[p].ch[d]=i,a[p].s++;
		for(int j=0;j<str[i]-'0';j++)
		{
			S[0].push(i);
			S[1].push(j);
		}
	}
	getf(1,0);getf(1,1);getf(1,2);
	getfs(1,0);getfs(1,1);getfs(1,2);
	for(int i=0;i<=2;i++)
		ans=max(ans,a[1].f[i]);
	printf("%d ",ans);
	ans=INF;
	for(int i=0;i<=2;i++)
		ans=min(ans,a[1].fs[i]);
	printf("%d\n",ans);
	return 0;
}