首页 > 代码库 > 2795: [Poi2012]A Horrible Poem

2795: [Poi2012]A Horrible Poem

2795: [Poi2012]A Horrible Poem

Time Limit: 50 Sec  Memory Limit: 128 MB
Submit: 484  Solved: 235
[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

HINT

 

Source

鸣谢 jiangzoi&oimaster

技术分享

 

技术分享

首先我们可以知道,一个长度为len的子串,它的循环节一定是len的约数,所以只要找len的约数,再用hash判断就行了。但是这样的复杂度是q√n的,会TLE,所以考虑优化。
考虑如果有一个字母出现了k次,那么这个子串的循环节个数一定是k的约数,我们把所有的k取一个gcd,再找约数。

#include<cstdio>#include<algorithm>#include<iostream>#define P 31#define R register#define ull unsigned long longusing namespace std;int read(){    R int x=0;bool f=1;    R char ch=getchar();    while(ch<0||ch>9){if(ch==-)f=0;ch=getchar();}    while(ch>=0&&ch<=9){x=x*10+ch-0;ch=getchar();}    return f?x:-x;}const int N=5e5+10;ull hash[N],base[N];int n,m,vgcd,ans,c[N][26];;char s[N];void get_hash(){    base[0]=1;    for(int i=1;i<=n;i++){        hash[i]=hash[i-1]*P+s[i]-a;        base[i]=base[i-1]*P;    }}void check(int x,int y,int t){    ull has1=hash[y-t]-hash[x-1]*base[y-x+1-t];    ull has2=hash[y]-hash[x+t-1]*base[y-x+1-t];    if(has1==has2) ans=min(ans,t);}int main(){    n=read();gets(s+1);m=read();    get_hash();    for(int i=0;i<=25;i++){        for(int j=1;j<=n;j++){            c[j][i]=c[j-1][i]+(s[j]-a==i);        }    }    for(int i=1,x,y;i<=m;i++){        x=read();y=read();ans=N;vgcd=y-x+1;        for(int j=0;j<=25;j++) vgcd=__gcd(vgcd,c[y][j]-c[x-1][j]);        for(int j=1;j*j<=vgcd;j++){            if(vgcd%j) continue;            check(x,y,(y-x+1)/j);            check(x,y,(y-x+1)/(vgcd/j));        }        printf("%d\n",ans);    }    return 0;}

 

 

 

2795: [Poi2012]A Horrible Poem