BZOJ4236 JOIOJI 【map】

BZOJ4236 JOIOJI

---

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\ast 10^5$

---

我们考虑怎么快速判断两个位置作差后JOI的数量相等
$J_i=J_i$,$O_i=O_j$,$I_i=I_i$
$J_i-O_i=J_j-O_j,O_i-I_i=O_j-I_j$
然后我们就可以用map来压缩每一个位置的值

然后就做完了

---

注意第0个位置也存在

---

#include<bits/stdc++.h>
using namespace std;
#define N 200010
#define fu(a,b,c) for(int a=b;a<=c;a++)
#define fd(a,b,c) for(int a=b;a>=c;a--)
#define pi pair<int,int>
int J=0,O=0,I=0,n;
pi p[N];
char c[N];
map<pi,int> mp;
int main(){
  scanf("%d%s",&n,c+1);
  mp[pi(0,0)]=0;
  fu(i,1,n){
    if(c[i]=='J')J++;
    if(c[i]=='O')O++;
    if(c[i]=='I')I++;
    p[i]=pi(I-O,O-J);
    if(!mp.count(p[i]))mp[p[i]]=i;
  }
  int ans=0;
  fd(i,n,1)ans=max(ans,i-mp[p[i]]);
  printf("%d",ans);
  return 0;
}