bzoj4236 JOIOJI hash 模拟

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

Hint

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

1<=N<=2*10^5

题解

　这道题目是有些巧妙的，一段连续的区间j,o,i三个字母出现次数相等。

　比如i行和j行之间是的，那么满足，J[j]-J[i]=O[j]-O[i]=I[j]-I[i]=x,

　那么可以得到，J[i]-O[i]=J[j]-O[j]，另外一组也是一样的，然后就是双HASH判断了。

 1 #include <map>
 2 #include <cstdio>
 3 #include <cstring>
 4 #include <iostream>
 5 #include <algorithm>
 6 #define M 200200
 7 using namespace std;
 8 int n,ans;
 9 char s[M];
10 map<pair<int,int>,int> m;
11 int sum[3][M];
12 int main()
13 {
14     int i;
15     cin>>n;
16     scanf("%s",s+1);
17     m[make_pair(0,0)]=0;
18     for(i=1;i<=n;i++)
19     {
20         sum[0][i]=sum[0][i-1]+(s[i]=='J');
21         sum[1][i]=sum[1][i-1]+(s[i]=='O');
22         sum[2][i]=sum[2][i-1]+(s[i]=='I');
23         if( m.find( make_pair(sum[0][i]-sum[1][i],sum[1][i]-sum[2][i]) )==m.end() )
24             m[make_pair(sum[0][i]-sum[1][i],sum[1][i]-sum[2][i])]=i;
25         else ans=max(ans,i-m[make_pair(sum[0][i]-sum[1][i],sum[1][i]-sum[2][i])]);
26     }
27     cout<<ans<<endl;
28 }