bzoj4236 JOIJOI

Posted 2020-06-18 ccz181078

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

计算出在长度为i的前缀中J,O,I分别出现的次数t1[i],t2[i],t3[i]

[l+1,r]符合要求等价于t1[r]-t1[l]=t2[r]-t2[l]=t3[r]-t3[l]

即t1[r]-t2[r]=t1[l]-t2[l]，且t1[r]-t3[r]=t1[l]-t3[l]

从左到右扫描字符串并把 (t1[i]-t2[i],t1[i]-t3[i]) → i 插入一个hashmap中，顺便判断以i结尾的最长合法长度

#include<cstdio>
#define N 200002
#define P 501257
typedef unsigned int U;
int n;
U t1[N],t2[N],t3[N];
char s[N];
U h1[P],h2[P],h3[P];
bool d[P];
int ans=0;
inline void ins(U a,U b,U x){
    U w=(a*13+b*5+97)%P;
    while(d[w]){
        if(h1[w]==a&&h2[w]==b){
            int v=x-h3[w];
            if(v>ans)ans=v;
            return;
        }
        w+=1237;
        if(w>=P)w-=P;
    }
    h1[w]=a;h2[w]=b;h3[w]=x;d[w]=1;
}
int main(){
    scanf("%d%s",&n,s+1);
    for(int i=1;i<=n;i++){
        char c=s[i];
        t1[i]=t1[i-1]+(c==‘J‘);
        t2[i]=t2[i-1]+(c==‘O‘);
        t3[i]=t3[i-1]+(c==‘I‘);
    }
    for(int i=0;i<=n;i++)ins(t1[i]-t2[i],t1[i]-t3[i],i);
    printf("%d",ans);
    return 0;
}