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

HINT

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

1<=N<=2*10^5

Solution

还以为是什么非常神仙的烤串题……我还撕烤了好久QAQ……

其实就是个乱搞，统计一下$sum[i][0/1/2]$也就是三种字母的前缀和。

记录一下二元组$(sum[i][0]-sum[i][1],sun[i][1]-sum[i][2])$最早出现的位置，然后开个$map$直接搞就好了……

记得一开始把$(0,0)$扔到$map$里去……

Code

 1 #include<cstdio>
 2 #include<map>
 3 #define N (200009)
 4 using namespace std;
 5 
 6 pair<int,int>p;
 7 int n,ans,sum[N][3];
 8 char s[N];
 9 map<pair<int,int>,int>Map;
10 int main()
11 {
12     scanf("%d%s",&n,s+1);
13     Map[make_pair(0,0)]=0;
14     for (int i=1; i<=n; ++i)
15     {
16         sum[i][0]=sum[i-1][0]+(s[i]=='J');
17         sum[i][1]=sum[i-1][1]+(s[i]=='O');
18         sum[i][2]=sum[i-1][2]+(s[i]=='I');
19         p=make_pair(sum[i][0]-sum[i][1],sum[i][1]-sum[i][2]);
20         if (Map.find(p)==Map.end()) Map[p]=i;
21         else ans=max(ans,i-Map[p]);
22     }
23     printf("%d
",ans);
24 }