BZOJ1864 [ZJOI2006] 三色二叉树

给出一颗二叉树，用红绿蓝进行染色，要求二叉树三角形的颜色不同，求绿色节点的数目的最大最小值。

给出树的方法比较特殊，因此直接递归即可，无须建图。

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

char str[1000005];
int pin;

int sw(int a,int b,int c){
    if(c) return max(a,b);
    else return min(a,b);
}

pair<int,int> dfs(int k){
    pin++;
    if(str[pin-1]=='0') return make_pair(1,0);
    if(str[pin-1]=='1'){
        pair<int,int> nxt = dfs(k);
        return make_pair(nxt.second+1, sw(nxt.first,nxt.second,k));
    }
    if(str[pin-1]=='2'){
        pair<int,int> cl = dfs(k);
        pair<int,int> cr = dfs(k);
        return make_pair(cl.second+cr.second+1, sw(cl.first+cr.second, cl.second+cr.first,k));
    }
}                

int main(){
    scanf("%s",&str);
    pair<int,int> result=dfs(1);
    printf("%d ",max(result.first, result.second));
    pin=0; result=dfs(0);
    printf("%d
",min(result.first, result.second));
}