BZOJ4236: JOIOJI

设ai为J在1~i出现的次数，bi为O在1~i出现的次数，ci为I在1~i出现的次数
对于l+1~r，若它满足题目要求，可得：
ar-al=br-bl=cr-cl
所以得
ar-br=al-bl
ar-cr=al-cl
br-cr=bl-cl
等式每边都只与一个位置的值有关
所以我们用一个三元组(ai-bi,ai-ci,bi-ci)代表i位置，用一个map维护，每个位置找之前有没有特征量相同的，若有即可尝试更新答案

#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
using namespace std;

void up(int &x,int y){if(x<y)x=y;}
const int maxn = 210000;

struct node
{
    int x,y;
    node(){}
    node(int _x,int _y){x=_x;y=_y;}
};
bool operator < (node x,node y){return x.x==y.x?x.y<y.y:x.x<y.x;}
map<node,int>H;

char st[maxn];
int n,m;

int main()
{
    scanf("%d",&n);
    scanf("%s",st+1);
    int a,b,c; a=b=c=0;
    int re=0;
    for(int i=1;i<=n;i++)
    {
        if(st[i]=='J')a++;
        else if(st[i]=='O')b++;
        else c++;

        node k=node(a-b,a-c);
        if(H.count(k)>0) up(re,i-H[k]);
        else H[k]=i;
    }
    if(a==b&&b==c) re=n;
    printf("%d\n",re);

    return 0;
}