BZOJ4384: [POI2015]Trzy wieże

可以先把只有一种字符的情况算出来
然后考虑没有任意两个字符数目相同的情况
考虑三种字母的前缀和x,y,z，然后差分
则当
X-x=Y-y
Y-y=Z-z
Z-z=X-x
这三个条件有一个满足时i+1~j的串不能贡献
移项得 (x-y,y-z,z-x)三元组
当三元互不相同时可以产生贡献
对第一维排序，树状数组以第二维为下标，维护前缀后缀的最大值，最大值的第三维，和最大值第三维不同的最大值
然后正反扫一遍就可以统计所有情况了

因为卡常代码加了一些剪枝..

code：

#include<set>
#include<map>
#include<deque>
#include<queue>
#include<stack>
#include<cmath>
#include<ctime>
#include<bitset>
#include<string>
#include<vector>
#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<climits>
#include<complex>
#include<iostream>
#include<algorithm>
#define ll long long
#define lowbit(x) x&(-x)
using namespace std;

inline void up(int &x,const int &y){if(x<y)x=y;}
inline void down(int &x,const int &y){if(x>y)x=y;}
const int maxn = 2400000;

int n,N;
struct A{int x,y,z,i;}a[maxn];
inline bool cmp(const A x,const A y){return x.x<y.x;}

struct node{int mx1,n,mx2;}pre[maxn],suf[maxn],re;
void clear(){for(int i=0;i<=N;i++) pre[i]=suf[i]=(node){0,0,0};}
inline void up(node &x,const node &y)
{
    if(x.n==y.n) up(x.mx1,y.mx1),up(x.mx2,y.mx2);
    else
    {
        if(x.mx1<=y.mx1) up(x.mx2,x.mx1>y.mx2?x.mx1:y.mx2),x.mx1=y.mx1,x.n=y.n;
        else up(x.mx2,y.mx1);
    }
}
void upd(node s[],int x,A &c){node nw=(node){c.i,c.z,-1};for(x;x<=N;x+=lowbit(x))up(s[x],nw);}
void que(node s[],int x){for(;x;x-=lowbit(x))up(re,s[x]);}

// B C S
char str[maxn];

int ans;

int main()
{
    scanf("%d",&n); N=n+1<<1; getchar();
    for(int i=1;i<=n;i++) str[i]=getchar();
    a[0].x=a[0].y=a[0].z=n+1; a[0].i=0;
    for(int i=1,now=0;i<=n;i++)
    {
        a[i]=a[i-1]; a[i].i=i;
        if(str[i]=='B') a[i].x++,a[i].z--;
        else if(str[i]=='C') a[i].x--,a[i].y++;
        else a[i].y--,a[i].z++;

        if(str[i]!=str[i-1]) now=0;
        ++now;
        ans=max(ans,now);
    }
    if(a[n].x!=n+1&&a[n].y!=n+1&&a[n].z!=n+1) {printf("%d\n",n); return 0;}
    sort(a,a+n+1,cmp);

    //clear();
    for(int i=0;i<=n;++i)
    {
        if(n-a[i].i>ans) {
        re=(node){N+1,a[i].z,-1};
        que(pre,a[i].y-1);
        up(ans,re.mx2-a[i].i);
        re=(node){N+1,a[i].z,-1};
        que(suf,N-a[i].y-1);
        up(ans,re.mx2-a[i].i); }

        if(a[i].x!=a[i+1].x)
        {
            for(int j=i;a[j].x==a[i].x;--j) if(a[j].i>ans)
                upd(pre,a[j].y,a[j]),
                upd(suf,N-a[j].y,a[j]);
        }
    }
    clear();
    for(int i=n;i>=0;--i)
    {
        if(n-a[i].i>ans) {
        re=(node){N+1,a[i].z,-1};
        que(pre,a[i].y-1);
        up(ans,re.mx2-a[i].i);
        re=(node){N+1,a[i].z,-1};
        que(suf,N-a[i].y-1);
        up(ans,re.mx2-a[i].i); }

        if(a[i].x!=a[i-1].x)
        {
            for(int j=i;a[j].x==a[i].x;++j) if(a[j].i>ans)
                upd(pre,a[j].y,a[j]),
                upd(suf,N-a[j].y,a[j]);
        }
    }
    printf("%d\n",ans);

    return 0;
}