BZOJ 1864: [Zjoi2006]三色二叉树 树形DP + 读入

Description

Input

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

Output

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

题解：本题大水题，主要卡读入. 能把树建出来就普及组水题了~

#include <bits/stdc++.h>
#define setIO(s) freopen(s".in","r",stdin)
#define maxn 1000000
#define lson ch[u][0]
#define rson ch[u][1]
using namespace std;
int ch[maxn][2], f[maxn], g[maxn];
int n = 1;
void read(int u)
{
    char c = getchar();
    if(c == '0') return;
    ++n, lson = n, read(lson);
    if(c == '2') ++n, rson = n, read(rson);
}
void dfs1(int u)
{
    if(!u) return;
    dfs1(lson), dfs1(rson);
    f[u] = g[lson] + g[rson] + 1;
    g[u] = max(f[lson] + g[rson], g[lson] + f[rson]);
}
void dfs2(int u)
{
    if(!u) return;
    dfs2(lson), dfs2(rson);
    f[u] = g[u] = 0;
    f[u] = g[lson] + g[rson] + 1;
    g[u] = min(f[lson] + g[rson], g[lson] + f[rson]);
}
int main()
{
    // setIO("input");
    read(1);
    dfs1(1);
    printf("%d ",max(g[1], f[1]));
    dfs2(1);
    printf("%d\n",min(g[1], f[1]));
    return 0;
}