[BZOJ4236]JOIOJI

设a[i],b[i],c[i]分别为前i个数中J,O,I的个数，则一个区间[L+1,R]合法当且进当：

a[r]-a[l]=b[r]-b[l]=c[r]-c[l]。

即a[l]-b[l]=a[r]-b[r],a[l]-c[l]=a[r]-c[r],b[l]-c[l]=b[r]-c[r]。

用map记录每个三元组<a[i],b[i],c[i]>第一次出现的位置，每次尝试更新答案。

#include<map>
#include<cstdio>
#include<algorithm>
#define rep(i,l,r) for (int i=(l); i<=(r); i++)
using namespace std;

const int N=200010;
int n,ans,sm1[N],sm2[N],sm3[N];
char s[N];
struct P{ int a,b,c; };
bool operator <(const P &a,const P &b)
{ return (a.a==b.a) ? (a.b==b.b ? a.c<b.c : a.b<b.b) : a.a<b.a; }
map<P,int>mp;

int main(){
    freopen("bzoj4236.in","r",stdin);
    freopen("bzoj4236.out","w",stdout);
    scanf("%d%s",&n,s+1); mp[(P){0,0,0}]=0;
    rep(i,1,n){
        sm1[i]=sm1[i-1]; sm2[i]=sm2[i-1]; sm3[i]=sm3[i-1];
        if (s[i]=='J') sm1[i]++;
        if (s[i]=='O') sm2[i]++;
        if (s[i]=='I') sm3[i]++;
        P t=(P){sm1[i]-sm2[i],sm1[i]-sm3[i],sm2[i]-sm3[i]};
        if (mp.count(t)) ans=max(ans,i-mp[t]); else mp[t]=i;
    }
    printf("%d\n",ans);
    return 0;
}