洛谷 2585 [ZJOI2006]三色二叉树——树形dp

题目：https://www.luogu.org/problemnew/show/P2585

可以把不是绿色的记成一种。仔细一想不会有冲突。如果自己是绿色，孩子的不同颜色不会冲突；如果自己不是绿色，自己的不是绿色的孩子对于自己就像二分图一样的感觉，所以总有方案使得不区分另外两种颜色也不会有冲突。

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
const int N=5e5+;
int n,rt,tot,ls[N],rs[N],dp[N][][],p0;//是/否绿 最大/小
char ch[N];
void build(int &cr,int dep)
{
    cr=++tot;
    if(ch[p0]=='')p0++;
    else if(ch[p0]=='')p0++,build(ls[cr],dep+);
    else
        p0++,build(ls[cr],dep+),build(rs[cr],dep+);
}
void dfs(int cr)
{
    if(!ls[cr])
    {
        dp[cr][][]=dp[cr][][]=;
        dp[cr][][]=dp[cr][][]=;
        return;
    }
    else if(!rs[cr])
    {
        int v=ls[cr]; dfs(v);
        dp[cr][][]=dp[v][][]+;
        dp[cr][][]=dp[v][][]+;
        dp[cr][][]=max(dp[v][][],dp[v][][]);
        dp[cr][][]=min(dp[v][][],dp[v][][]);
    }
    else
    {
        int Ls=ls[cr],Rs=rs[cr];
        dfs(Ls); dfs(Rs);
        dp[cr][][]=dp[Ls][][]+dp[Rs][][]+;
        dp[cr][][]=dp[Ls][][]+dp[Rs][][]+;
        dp[cr][][]=max(dp[Ls][][]+dp[Rs][][],
                         dp[Ls][][]+dp[Rs][][]);
        dp[cr][][]=min(dp[Ls][][]+dp[Rs][][],
                         dp[Ls][][]+dp[Rs][][]);
    }
}
int main()
{
    //freopen("TRO.IN","r",stdin);
    //freopen("TRO.OUT","w",stdout);
    scanf("%s",ch);n=strlen(ch);
    build(rt,);
    dfs(rt);
    printf("%d %d\n",max(dp[rt][][],dp[rt][][]),
           min(dp[rt][][],dp[rt][][]));
    return ;
}