[bzoj1864][Zjoi2006]三色二叉树（树上dp）

#include<cstdio>
#include<algorithm>
#include<cmath>
#include<cstdlib>
#include<iostream>
#include<cstring>
const int N=500001;
using namespace std;
int n,l[N],r[N],f[N][2];
void build(int x)
{
char ch=getchar();
if (ch=='0') return;
build(l[x]=++n);
if (ch=='2') build(r[x]=++n);
}

void dp(int x,int y)
{
if (!x) return;
dp(l[x],y);
dp(r[x],y);
f[x][1]=f[l[x]][0]+f[r[x]][0]+1;
f[x][0]=y?min(f[l[x]][0]+f[r[x]][1],f[l[x]][1]+f[r[x]][0]):max(f[l[x]][0]+f[r[x]][1],f[l[x]][1]+f[r[x]][0]);
}

int main()
{
build(n=1);
dp(1,0);
printf("%d ",max(f[1][0],f[1][1]));
dp(1,1);
printf("%d",min(f[1][0],f[1][1]));
}