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”三个字母的连续子串。

1<=N<=2*10^5

Solution

---

设s1,s2,s3分别表示字母J,O,I出现次数的前缀和
满足s1’所在位置到s1所在位置是一个可行解一定有s1-s1’=s2-s2’=s3-s3’
移项一下可以得到s1-s2=s1’-s2’，可以发现等式左边和右边都只和所在位置相关，用map记录一下二元组所在的位置即可

注意答案可能本身就是前缀的情况，具体实现可以看代码

Code

---

#include <stdio.h>
#include <string.h>
#include <algorithm>
#include <map>
#define rep(i,st,ed) for (int i=st;i<=ed;++i)

typedef std:: pair <int, int> pair;

std:: map <pair, int> map;

char str[200005];

int main(void) {
    int n; scanf("%d",&n);
    scanf("%s",str+2); map[pair(0,0)]=1;
    int s1=0,s2=0,s3=0,ans=0;
    rep(i,2,n+1) {
        if (str[i]=='J') s1++;
        if (str[i]=='O') s2++;
        if (str[i]=='I') s3++;
        int pos=map[pair(s1-s2,s2-s3)];
        if (pos) ans=std:: max(ans,i-pos);
        else map[pair(s1-s2,s2-s3)]=i;
    }
    printf("%d\n", ans);
    return 0;
}