【BZOJ1864】【ZJOI2006】三色二叉树（树形dp）

Description

Input

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

Output

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

Sample Input

1122002010

Sample Output

5 2

题解：
f[i][0/1]表示i结点为绿/非绿色的绿色结点的最大个数
转移
f[x][1]=f[l[x]][0]+f[r[x]][0]+1;
f[x][0]=max/min(f[l[x]][0]+f[r[x]][1],f[r[x]][0]+f[l[x]][1]);

#include<iostream>
#include<stdio.h>
#include<algorithm>
#include<string.h>
#include<math.h>
#define ll long long
#define inf 0x7f7f7f7f
#define N 300005
using namespace std;
int cnt=1,l[N],r[N],f[N][2];
void buildt(int u)
{
    char c=getchar();
    if(c=='0') return;
    l[u]=++cnt;buildt(cnt);
    if(c=='2')
    {
        r[u]=++cnt;
        buildt(cnt);
    }
}
void dp(int u,bool fla)
{
    if(!u) return;
    dp(l[u],fla),dp(r[u],fla);
    f[u][1]=f[l[u]][0]+f[r[u]][0]+1;
    if(fla) f[u][0]=max(f[l[u]][0]+f[r[u]][1],f[r[u]][0]+f[l[u]][1]);
    else f[u][0]=min(f[l[u]][0]+f[r[u]][1],f[r[u]][0]+f[l[u]][1]);
}
int main()
{
    buildt(1);
    dp(1,1);
    printf("%d ",max(f[1][1],f[1][0]));
    memset(f,0,sizeof(f));
    dp(1,0);
    printf("%d\n",min(f[1][1],f[1][0]));
    return 0;
}

感觉这题的读入挺有意思的。