bzoj1864 三色二叉树 树形dp

       把字符串转化成树,然后f[i][0],f[i][1],f[i][2]表示i取绿色,蓝色,红色时所在子树的最大值,同理g[i][0..2]表示最小值。然后跑树形dp即可。

AC代码如下:

#include<iostream>
#include<cstdio>
#include<cstring>
#define N 500005
using namespace std;

int cnt,rt,f[N][3],g[N][3],c[N][2]; char s[N];
void dfs(int &k){
	k=++cnt; int t=s[cnt]-'0';
	if (t) dfs(c[k][0]); if (t>1) dfs(c[k][1]);
}
void dp(int k){
	if (!k) return;
	int l=c[k][0],r=c[k][1],i; dp(l); dp(r);
	for (i=0; i<3; i++){
		int x=(i+1)%3,y=(x+1)%3,z=(i)?0:1;
		f[k][i]=max(f[l][x]+f[r][y],f[l][y]+f[r][x])+z;
		g[k][i]=min(g[l][x]+g[r][y],g[l][y]+g[r][x])+z;
	}
}
int main(){
	scanf("%s",s+1); dfs(rt); dp(1);
	printf("%d %d\n",max(f[1][0],max(f[1][1],f[1][2])),min(g[1][0],min(g[1][1],g[1][2])));
	return 0;
}


by lych

2016.2.26

你可能感兴趣的:(dp,DFS,树形DP)