【bzoj4236】【JOIJOI】【map】

Description

JOIOJI桑是JOI君的叔叔。“JOIOJI”这个名字是由“J、O、I”三个字母各两个构成的。

最近，JOIOJI桑有了一个孩子。JOIOJI桑想让自己孩子的名字和自己一样由“J、O、I”三个字母构成，并且想让“J、O、I”三个字母的出现次数恰好相同。

JOIOJI桑家有一份祖传的卷轴，上面写着一首长诗，长度为N，由“J、O、I”三个字母组成。JOIOJIさん想用诗中最长的满足要求的连续子串作为孩子的名字。

现在JOIOJI桑将这首长诗交给了你，请你求出诗中最长的、包含同样数目的“J、O、I”三个字母的连续子串。

Input

第一行一个正整数N，代表这首长诗的长度

接下来一行一个长度为N的字符串S，表示这首长诗，保证每个字符都是“J、O、I”三个字母中的一个

Output

输出一行一个正整数，代表最长的包含等数量“J、O、I”三个字母的最长连续子串的长度。如果不存在这样的子串，输出0

Sample Input

10
JOIIJOJOOI

Sample Output

6

HINT

选择“IIJOJO”这个子串，长度为6，包含“J、O、I”三个字母各2个，这是最长的满足要求的子串。

1<=N<=2*10^5

题解：用f[i][0],f[i][1],f[i][2]分别表示到第i为J,O,I出现的次数，

用map保存一下当J和O的数量差为a,O和I的数量差为b时,最靠前的位置。

每次查询一下,更新答案和map即可。

代码：

#include<iostream>
#include<cstdio>
#include<map>
#include<cstring>
#define N 200010 
using namespace std;
int f[N][5],n,ans,len;
char ch[N];
struct use{
    int a,b;
    bool operator<(const use t)const{if (a==t.a) return b<t.b;else return a<t.a;} 
};
map<use,int>q;
int main(){
   scanf("%d",&n);scanf("%s",ch);len=strlen(ch);
   q[(use){0,0}]=-1;
   for (int i=0;i<len;i++){
     f[i+1][0]=f[i][0];f[i+1][1]=f[i][1];f[i+1][2]=f[i][2];
     if (ch[i]=='J') f[i+1][0]++;
     if (ch[i]=='O') f[i+1][1]++;
     if (ch[i]=='I') f[i+1][2]++;
     if (!q[(use){f[i+1][0]-f[i+1][1],f[i+1][1]-f[i+1][2]}]) 
      q[(use){f[i+1][0]-f[i+1][1],f[i+1][1]-f[i+1][2]}]=i;
     else ans=max(ans,i-q[(use){f[i+1][0]-f[i+1][1],f[i+1][1]-f[i+1][2]}]);
   }    
  cout<<ans<<endl;
}