【BZOJ】4236 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

Solution

这个问题可以用前缀和相减的思想，那么也就是要求两个前缀 <J1,O1,I1><J2,O2,I2> <script type="math/tex" id="MathJax-Element-277"> </script>，使得 $j_2-j_1=O_2-O_1=I_2-I_1$
那么我们再做一下差……就是求 <O1−J1,I1−J1>=<O2−J2,I2−J2> <script type="math/tex" id="MathJax-Element-279"> = </script>
这个东西……塞map硬上（貌似对pascal很不友好啊2333）

#include<stdio.h>
#include<map>

using namespace std;

map<pair<int,int>,int> m;
int n,n1,n2,n3,ans;
char ch;

int main()
{
    scanf("%d\n",&n);m[make_pair(0,0)]=0;
    for (int i=1;i<=n;i++)
    {
        ch=getchar();if (ch=='J') n1++;
        if (ch=='O') n2++;if (ch=='I') n3++;
        if (m.find(make_pair(n2-n1,n3-n1))!=m.end()) ans=max(ans,i-m[make_pair(n2-n1,n3-n1)]);
        else m[make_pair(n2-n1,n3-n1)]=i;
    }
    printf("%d",ans);
}