BZOJ1864 [ZJOI2006] 三色二叉树

题目链接：http://www.lydsy.com/JudgeOnline/problem.php?id=1864

Description

Input

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

Output

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

树DP

用f[x][0]表示编号为x的点不染成绿色，f[x][1]表示编号为x的点染成绿色

f[x][0] = max (或min) (f[l[x]][0] + f[r[x]][1], f[r[x]][0] + f[l[x]][1]);
f[x][1] = f[l[x]][0] + f[r[x]][0] + 1;

#include <cstdio>
#include <algorithm>
#include <cstring>
#define rep(i,l,r) for(int i=l; i<=r; i++)
#define clr(x,y) memset(x,y,sizeof(x))
using namespace std;
const int INF = 0x3f3f3f3f;
const int maxn = 500010;
int tot=1,l[maxn],r[maxn],f[maxn][2];
void dfs(int x){
    char ch = getchar();
    if (ch == '0') return;
    tot++; l[x] = tot; dfs(tot);
    if (ch == '2'){
        tot++; r[x] = tot; dfs(tot);
    }
}
void dp1(int x){
    if (!x) return;
    dp1(l[x]); dp1(r[x]);
    f[x][0] = max(f[l[x]][0] + f[r[x]][1], f[r[x]][0] + f[l[x]][1]);
    f[x][1] = f[l[x]][0] + f[r[x]][0] + 1;
}
void dp2(int x){
    if (!x) return;
    dp2(l[x]); dp2(r[x]);
    f[x][0] = min(f[l[x]][0] + f[r[x]][1], f[r[x]][0] + f[l[x]][1]);
    f[x][1] = f[l[x]][0] + f[r[x]][0] + 1;
}
int main(){
    clr(l,0); clr(r,0); dfs(1);
    clr(f,0); dp1(1); printf("%d ",max(f[1][0],f[1][1]));
    clr(f,0); dp2(1); printf("%d
",min(f[1][0],f[1][1]));
    return 0;
}