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 Schar 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;}