一道树形DP入门题,不需要多叉树转二叉树。
用f(i,j)表示第i个节点涂j颜色时,其子树顶点为绿色最多的个数,fs(i,j)则表示最少的个数。很容易写出状态转移方程,用记忆化搜索实现。
然后对于输入,可以利用DFS或栈来转化成一棵树。
貌似这个代码还可以继续精简,没必要这么长。
//BZOJ1864
#include<iostream>
#include<cstdio>
#include<algorithm>
#include<string>
#include<cstring>
#include<queue>
#include<vector>
#include<stack>
//Red 0 Green 1 Blue 2
using namespace std;
const int MAXN=500010;
const int INF=1e9;
struct Node
{
int ch[2],s,f[3],fs[3];
Node() {f[0]=f[1]=f[2]=-1;fs[0]=fs[1]=fs[2]=-1;}
}a[MAXN];
stack<int> S[2];
char str[MAXN];
int len,ans;
int getf(int x,int c)
{
if(a[x].f[c]!=-1) return a[x].f[c];
if(a[x].s==0)
{
if(c==1) a[x].f[c]=1;
else a[x].f[c]=0;
return a[x].f[c];
}
if(a[x].s==1)
{
for(int i=0;i<=2;i++)
if(i!=c) a[x].f[c]=max(a[x].f[c],getf(a[x].ch[0],i));
if(c==1) a[x].f[c]++;
return a[x].f[c];
}
if(a[x].s==2)
{
int l,r;
for(int i=0;i<=2;i++) if(i!=c) {l=i;break;}
for(int i=0;i<=2;i++) if(i!=c&&i!=l) {r=i;break;}
a[x].f[c]=max(a[x].f[c],getf(a[x].ch[0],l)+getf(a[x].ch[1],r));
a[x].f[c]=max(a[x].f[c],getf(a[x].ch[0],r)+getf(a[x].ch[1],l));
if(c==1) a[x].f[c]++;
return a[x].f[c];
}
}
int getfs(int x,int c)
{
if(a[x].fs[c]!=-1) return a[x].fs[c];
if(a[x].s==0)
{
if(c==1) a[x].fs[c]=1;
else a[x].fs[c]=0;
return a[x].fs[c];
}
if(a[x].s==1)
{
a[x].fs[c]=INF;
for(int i=0;i<=2;i++)
if(i!=c) a[x].fs[c]=min(a[x].fs[c],getfs(a[x].ch[0],i));
if(c==1) a[x].fs[c]++;
return a[x].fs[c];
}
if(a[x].s==2)
{
int l,r;
for(int i=0;i<=2;i++) if(i!=c) {l=i;break;}
for(int i=0;i<=2;i++) if(i!=c&&i!=l) {r=i;break;}
a[x].fs[c]=INF;
a[x].fs[c]=min(a[x].fs[c],getfs(a[x].ch[0],l)+getfs(a[x].ch[1],r));
a[x].fs[c]=min(a[x].fs[c],getfs(a[x].ch[0],r)+getfs(a[x].ch[1],l));
if(c==1) a[x].fs[c]++;
return a[x].fs[c];
}
}
int main()
{
scanf("%s",str+1);
len=strlen(str+1);
for(int i=0;i<str[1]-'0';i++)
{
S[0].push(1);
S[1].push(i);
}
for(int i=2;i<=len;i++)
{
int p=S[0].top(),d=S[1].top();
S[0].pop();S[1].pop();
a[p].ch[d]=i,a[p].s++;
for(int j=0;j<str[i]-'0';j++)
{
S[0].push(i);
S[1].push(j);
}
}
getf(1,0);getf(1,1);getf(1,2);
getfs(1,0);getfs(1,1);getfs(1,2);
for(int i=0;i<=2;i++)
ans=max(ans,a[1].f[i]);
printf("%d ",ans);
ans=INF;
for(int i=0;i<=2;i++)
ans=min(ans,a[1].fs[i]);
printf("%d\n",ans);
return 0;
}