[BZOJ1864][Zjoi2006]三色二叉树-动态规划

三色二叉树

Description

Input

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

Output

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

Sample Input

1122002010

Sample Output

5 2

HINT

---

这是浙江省选的难度……
那最近这次浙江省选是什么玩意…..
根本不是一个难度的好吗……

---

思路:
咱发现交换左右子树对答案无影响~
然后就可以默认优先左子树~
然后令dp[i][j]代表i号点涂j颜色的最大方案数~
然后，然后就没有然后了，一个简单的递推解决~

#include<iostream>
#include<cstdio>
#include<cstdlib>
#include<algorithm>

using namespace std;

const int N=1000233;

int ls[N],rs[N],v[N],ty;
int dp[N][5],n;

void read(int pos)
{
    v[pos]=getchar()-'0';
    if(v[pos])
    {
        read(ls[pos]=++n);
        if(v[pos]==2)
            read(rs[pos]=++n);
    }
}

inline int getval(int a,int b)
{
    if(!ty)return a>b?a:b;
    return a<b?a:b;
}

inline int work()
{
    for(int i=n;i>=1;i--)
        for(int c=0;c<3;c++)
        {
            if(v[i]==0)
            {
                dp[i][c]=c==0;
                continue;
            }

            if(!ty)
                dp[i][c]=0;
            else
                dp[i][c]=N;

            if(v[i]==2)
            {
                if(c==0)
                    dp[i][0]=getval(dp[ls[i]][1]+dp[rs[i]][2],dp[ls[i]][2]+dp[rs[i]][1])+1;
                if(c==1)
                    dp[i][1]=getval(dp[ls[i]][0]+dp[rs[i]][2],dp[ls[i]][2]+dp[rs[i]][0]);
                if(c==2)
                    dp[i][2]=getval(dp[ls[i]][0]+dp[rs[i]][1],dp[ls[i]][1]+dp[rs[i]][0]);
            }
            else
            {
                if(c!=0)
                    dp[i][c]=getval(dp[i][c],dp[ls[i]][0]);
                if(c!=1)
                    dp[i][c]=getval(dp[i][c],dp[ls[i]][1]);
                if(c!=2)
                    dp[i][c]=getval(dp[i][c],dp[ls[i]][2]);
                if(c==0)
                    dp[i][c]++;
            }
        }
}

int main()
{
    int ans1=0,ans2=N;
    read(++n);

    work();
    for(int i=0;i<3;i++)
        ans1=max(ans1,dp[1][i]);

    ty=1;
    work();
    for(int i=0;i<3;i++)
        ans2=min(ans2,dp[1][i]);

    printf("%d %d\n",ans1,ans2);

    return 0;
}