bzoj 2795 [Poi2012]A Horrible Poem hash+数论

2795: [Poi2012]A Horrible Poem

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

 %%https://blog.csdn.net/ww140142/article/details/48260615

#include<cstring>
#include<cmath>
#include<cstdio>
#include<algorithm>
#include<iostream>
#include<vector>

#define G 29
#define N 500007
#define mod 1000000007
#define ll long long

#define Wb putchar(' ')
#define We putchar('
')
#define rg register int
using namespace std;
inline int read()
{
    int x=0,f=1;char ch=getchar();
    while(!isdigit(ch)){if(ch=='-')f=-1;ch=getchar();}
    while(isdigit(ch)){x=(x<<1)+(x<<3)+ch-'0';ch=getchar();}
    return x*f;
}
inline void write(int x)
{
    if(x<0) putchar('-'),x=-x;
    if (x==0) putchar(48);
    int num=0;char c[15];
    while(x) c[++num]=(x%10)+48,x/=10;
    while(num) putchar(c[num--]);
}

int n,Q;
char ch[N];
ll bin[N],hash[N];
bool flag[N];
vector<int>ve[N];

void init_prime()
{
    for (rg i=2;i<=n;i++)
        if (!flag[i])
        {
            rg x=i;
            while(x<=n)
            {
                int z=x;
                while(z%i==0) ve[x].push_back(i),z/=i;
                flag[x]=true;
                x+=i;
            }
        }
}
bool judge(int x,int l,int r)
{
    ll res1=hash[r-x]-(hash[l-1]*bin[(r-x)-l+1])%mod;
    ll res2=hash[r]-(hash[l+x-1]*bin[(r-x)-l+1])%mod;
    res1=(res1%mod+mod)%mod,res2=(res2%mod+mod)%mod;
    return res1==res2;
}
int main()
{
    n=read(),scanf("%s",ch+1),bin[0]=1,init_prime();
    for (int i=1;i<=n;i++)
        hash[i]=(hash[i-1]*G+ch[i]-'a')%mod,bin[i]=(bin[i-1]*G)%mod;
    Q=read();
    while(Q--)
    {
        int l=read(),r=read(),num=r-l+1,ans=num;
        if (num==1) write(1),We;
        else 
        {
            for (rg i=0;i<ve[num].size();i++)
                if (judge(ans/ve[num][i],l,r)) ans/=ve[num][i];
            write(ans),We;
        }
    }
}