BZOJ4384: [POI2015]Trzy wieże 记忆化搜索

#include<cstdio>
#include<algorithm>
#define gm 1000005
using namespace std;
int n,ans,pre[gm],suf[gm],dp[gm][3];
char s[gm];
int dfs(int pos,int a,int b,int c)
{
if(pos<=1) return 1;
if(a==b&&b==c&&a==c)
{
if(s[pos]=='S') goto S;
if(s[pos]=='B') goto B;
if(s[pos]=='C') goto C;
}
if(a==b)
{
if(s[pos]=='S'||s[pos]=='B')
{
if(a!=c+1) return pos-1;
if(pre[pos]>1) return pos-2;
if(dp[pos][0]) return dp[pos][0];
return dp[pos][0]=dfs(pos-1,a-(s[pos]=='S'),b-(s[pos]=='B'),c);
}
C: return dfs(pos-pre[pos],a,b,c-pre[pos]);
}
if(b==c)
{
if(s[pos]=='B'||s[pos]=='C')
{
if(b!=a+1) return pos-1;
if(pre[pos]>1) return pos-2;
if(dp[pos][1]) return dp[pos][1];
return dp[pos][1]=dfs(pos-1,a,b-(s[pos]=='B'),c-(s[pos]=='C'));
}
S: return dfs(pos-pre[pos],a-pre[pos],b,c);
}
if(a==c)
{
if(s[pos]=='S'||s[pos]=='C')
{
if(c!=b+1) return pos-1;
if(pre[pos]>1) return pos-2;
if(dp[pos][2]) return dp[pos][2];
return dp[pos][2]=dfs(pos-1,a-(s[pos]=='S'),b,c-(s[pos]=='C'));
}
B: return dfs(pos-pre[pos],a,b-pre[pos],c);
}
return pos;
}
int m
4000
ain()
{
scanf("%d",&n); fread(s,1,n+1,stdin); int a=0,b=0,c=0;
for(int i=1;i<=n;++i) pre[i]=s[i-1]==s[i]?pre[i-1]+1:1,++(s[i]=='B'?b:s[i]=='C'?c:a);
for(int i=n;i;--i) suf[i]=s[i+1]==s[i]?suf[i+1]+1:1;
for(int i=1;i<=n;++i)
ans=max(ans,max(suf[i],dfs(n,a,b,c)-i+1)),--(s[i]=='B'?b:s[i]=='C'?c:a);
printf("%d\n",ans);
return 0;
}