【bzoj 2795】A Horrible Poem(字符串哈希)

傳送門biu~
如果一個長度爲l的串s的循環節爲n,那麼需要滿足n是l的約數且s[1,l-n]=s[1+n,l]。
所以對於每一個詢問,枚舉每個質因子,用哈希判斷字符是否相等。

#include<bits/stdc++.h>
#define N 500005
#define base 19260817
using namespace std;
int n,q,prime[N],check[N],Hash[N],Pow[N],tp;char s[N];
inline void GetPrime(){
    for(int i=2;i<=n;++i){
        if(!check[i])   check[i]=prime[++tp]=i;
        for(int j=1;j<=tp && 1ll*i*prime[j]<=n;++j){
            check[i*prime[j]]=prime[j];
            if(i%prime[j]==0)   break;
        }
    }
}
inline bool eq(int l,int r,int L,int R){
    return (Hash[r]-Hash[l-1]*Pow[r-l+1])==(Hash[R]-Hash[L-1]*Pow[R-L+1]);
}
int main(){
    scanf("%d%s",&n,s+1);
    for(int i=1;i<=n;++i)    Hash[i]=Hash[i-1]*base+(s[i]-'a'+17);
    GetPrime(); Pow[0]=1;
    for(int i=1;i<=n;++i)    Pow[i]=Pow[i-1]*base;
    scanf("%d",&q);
    for(int l,r,len,i=1;i<=q;++i){
        scanf("%d%d",&l,&r); len=r-l+1;
        for(int j=len;j>1;){
            int p=check[j];
            while(len%p==0 && eq(l,r-len/p,l+len/p,r))  len/=p;
            while(j%p==0)   j/=p;
        }
        printf("%d\n",len);
    }
    return 0;
}

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章