BZOJ 1864: [Zjoi2006]三色二叉树 树形dp

1864: [Zjoi2006]三色二叉树

Time Limit: 1 Sec Memory Limit: 64 MB
Submit: 1065 Solved: 778
[Submit][Status][Discuss]

Description

Input

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

Output

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

Sample Input

1122002010

Sample Output

5 2

HINT

Source

Day1

题解：
树形dp乱搞。
难点在读入吧qaq
其实第二维只用0/1记录是否是绿色就可以了，我用了0,1,2…

以后碰到这种读入递归来搞啊！！递归白学了吗真的是……

#include<bits/stdc++.h>
using namespace std;

const int N = 500000 + 10;

int n;
int l[N],r[N],cnt=1;

void read(int u){
    char ch=getchar();
    if(ch=='0')return;
    else if(ch=='1'){
        ++cnt;l[u]=cnt;read(cnt);
    }
    else{
        ++cnt;l[u]=cnt;read(cnt);
        ++cnt;r[u]=cnt;read(cnt);
    }
}

int dp1[N][3],dp2[N][3];
void dfs(int u){
    if(u==0) return ;
    dfs(l[u]);dfs(r[u]);
    int v1=l[u],v2=r[u];

    dp1[u][0]+=max(dp1[v1][1]+dp1[v2][2],dp1[v1][2]+dp1[v2][1])+1;
    dp1[u][2]+=max(dp1[v1][0]+dp1[v2][1],dp1[v2][0]+dp1[v1][1]);
    dp1[u][1]+=max(dp1[v1][0]+dp1[v2][2],dp1[v2][0]+dp1[v1][2]);

    dp2[u][0]+=min(dp2[v1][1]+dp2[v2][2],dp2[v1][2]+dp2[v2][1])+1;
    dp2[u][2]+=min(dp2[v1][0]+dp2[v2][1],dp2[v2][0]+dp2[v1][1]);
    dp2[u][1]+=min(dp2[v1][0]+dp2[v2][2],dp2[v2][0]+dp2[v1][2]);
}

int main(){
    /*scanf("%s",s);n=strlen(s);
    for(int i=0;i<n;++i){
        if(s[i]=='0') continue;
        l[i+1]=++cnt;
        if(s[i]=='2') r[i+1]=++cnt;
    }*/
    /*
       wrong readin.
       sample :2221020000
    */
    read(1);
    dfs(1);
    printf("%d %d\n",max(max(dp1[1][1],dp1[1][2]),dp1[1][0]),min(min(dp2[1][1],dp2[1][2]),dp2[1][0]));
    return 0;
}