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

1864: [Zjoi2006]三色二叉树

Time Limit: 1 Sec   Memory Limit: 64 MB
[ Submit][ Status][ Discuss]

Description

Input

仅有一行，不超过500000个字符，表示一个二叉树序列。

Output

输出文件也只有一行，包含两个数，依次表示最多和最少有多少个点能够被染成绿色。

Sample Input

1122002010

Sample Output

5 2

HINT

Source

少看了个条件，很水，g求最小，f求最大，就完了

#include<cstdio>
#include<iostream>
using namespace std;
const int N = 500001 + 5;
char s[N];
int f[N][3],g[N][3],id;
void dfs(){
	id++;
	if( s[id] == '0' ){
		f[id][0] = 1; f[id][1] = f[id][2] = 0;
		g[id][0] = 1; g[id][1] = g[id][2] = 0;
	}
	if( s[id] == '1' ){
		int x = id; dfs();
		f[x][0] = max(f[x+1][1],f[x+1][2])+1; f[x][1]=max(f[x+1][0],f[x+1][2]); f[x][2]=max(f[x+1][0],f[x+1][1]);
		g[x][0] = min(g[x+1][1],g[x+1][2])+1; g[x][1]=min(g[x+1][0],g[x+1][2]); g[x][2]=min(g[x+1][0],g[x+1][1]);
	}
	if( s[id] == '2' ){
		int x = id; int l = id+1; dfs(); int r = id+1; dfs();
		f[x][0] = max(f[l][1]+f[r][2],f[l][2]+f[r][1])+1;
		f[x][1] = max(f[l][0]+f[r][2],f[l][2]+f[r][0]);
		f[x][2] = max(f[l][0]+f[r][1],f[l][1]+f[r][0]);
		g[x][0] = min(g[l][1]+g[r][2],g[l][2]+g[r][1])+1;
		g[x][1] = min(g[l][0]+g[r][2],g[l][2]+g[r][0]);
		g[x][2] = min(g[l][0]+g[r][1],g[l][1]+g[r][0]);
	}
}
int main(){
	scanf("%s", s+1);
	dfs();
	printf("%d %d", max(f[1][0],max(f[1][1],f[1][2])), min(g[1][0],min(g[1][1],g[1][2])));
	return 0;
}