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

http://www.lydsy.com/JudgeOnline/problem.php?id=4384
dp数组表示的是当前有两个数量相等，末尾字符是这两个中的一个且与它前面的字符不等，第三种的数量比这两个少1的情况。主要是基本相同的代码抄三遍所以看着比较长。。。时间复杂度o(n)。

#include<cstdio>#include<algorithm>#define gm 1000005using 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 main(){    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;}