bzoj 4236: JOIOJI【前缀和+map】

设sj,so,si分别是J O I的个数前缀和，然后要求求最长(l,r)满足sj[r]-sj[l-1]so[r]-so[l-1]si[r]-si[l-1]，化简一下就是满足so[r]-so[l-1]-(sj[r]-sj[l-1)-(si[r]-si[l-1]-(so[r]-so[l-1))==0，把这个搞成一个pair用map记录最早出现位置然后更新ans即可

注意因为是前缀和相减，所以0位置也要加进map

#include<iostream>
#include<cstdio>
#include<map>
using namespace std;
const int N=500005;
int n,ha[305],s[N][5],ans;
char c[N];
map<pair<int,int>,int>mp;
int main()
{
    scanf("%d%s",&n,c+1);
    ha['J']=1,ha['O']=2,ha['I']=3;
    for(int i=1;i<=n;i++)
    {
        for(int j=1;j<=3;j++)
            s[i][j]=s[i-1][j];
        s[i][ha[c[i]]]++;
    }
	mp[make_pair(0,0)]=1;
    for(int i=1;i<=n;i++)
    {
        if(mp[make_pair(s[i][2]-s[i][1],s[i][3]-s[i][2])])
            ans=max(ans,i-mp[make_pair(s[i][2]-s[i][1],s[i][3]-s[i][2])]+1);
        else
            mp[make_pair(s[i][2]-s[i][1],s[i][3]-s[i][2])]=i+1;
    }
    printf("%d\n",ans);
    return 0;
}