POI2012 AHorriblePoem解析-优快云博客

2795: [Poi2012]A Horrible Poem

Time Limit: 50 Sec Memory Limit:128 MB
Submit: 561 Solved: 276
[Submit][Status][Discuss]

Description

给出一个由小写英文字母组成的字符串S，再给出q个询问，要求回答S某个子串的最短循环节。
如果字符串B是字符串A的循环节，那么A可以由B重复若干次得到。

Input

第一行一个正整数n (n<=500,000)，表示S的长度。
第二行n个小写英文字母，表示字符串S。
第三行一个正整数q (q<=2,000,000)，表示询问个数。
下面q行每行两个正整数a,b (1<=a<=b<=n)，表示询问字符串S[a..b]的最短循环节长度。

Output

依次输出q行正整数，第i行的正整数对应第i个询问的答案。

Sample Input

8
aaabcabc
3
1 3
3 8
4 8

Sample Output

1
3
5

#include<iostream>
#include<cstdlib>
#include<cstdio>
#include<cstring>
#include<string>
using namespace std;
const int N=500005;
const int P=29;
char ch[N];
int n,minp[N],pri[N],len,t1,t2,m,x,y,ans,zs,p[N],has[N];
inline void prep(){
    for (int i=2;i<=n;++i){
        if (!minp[i])pri[++len]=(minp[i]=i);
        for (int j=1;j<=len && i*pri[j]<=n;++j){
            minp[i*pri[j]]=pri[j];
            if (i%pri[j]==0)break;
        }
    }
}
inline bool judge(int l1,int r1,int l2,int r2){
    t1=has[r1]-has[l1-1]*p[r1-l1+1];
    t2=has[r2]-has[l2-1]*p[r2-l2+1];
    return (t1==t2);
}
int main (){
    scanf ("%d%s",&n,ch+1);
    p[0]=1;
    for (int i=1;i<=n;++i){
        p[i]=p[i-1]*P;
        has[i]=has[i-1]*P+ch[i];
    }
    prep();scanf ("%d",&m);
    for (int i=1,j;i<=m;++i){
        scanf ("%d%d",&x,&y);
        for (j=ans=y-x+1;j>1;){
            zs=minp[j];
            while (ans%zs==0 && judge(x,y-ans/zs,x+ans/zs,y))ans/=zs;
            while (j%zs==0)j/=zs;
        }
        printf ("%d\n",ans);
    }
    return 0;
}