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('\n')
 #define rg register int
 using namespace std;
 inline int read()
 {
     int x=,f=;char ch=getchar();
     while(!isdigit(ch)){if(ch=='-')f=-;ch=getchar();}
     while(isdigit(ch)){x=(x<<)+(x<<)+ch-'';ch=getchar();}
     return x*f;
 }
 inline void write(int x)
 {
     if(x<) putchar('-'),x=-x;
     if (x==) putchar();
     int num=;char c[];
     while(x) c[++num]=(x%)+,x/=;
     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=;i<=n;i++)
         if (!flag[i])
         {
             rg x=i;
             while(x<=n)
             {
                 int z=x;
                 while(z%i==) 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-]*bin[(r-x)-l+])%mod;
     ll res2=hash[r]-(hash[l+x-]*bin[(r-x)-l+])%mod;
     res1=(res1%mod+mod)%mod,res2=(res2%mod+mod)%mod;
     return res1==res2;
 }
 int main()
 {
     n=read(),scanf("%s",ch+),bin[]=,init_prime();
     for (int i=;i<=n;i++)
         hash[i]=(hash[i-]*G+ch[i]-'a')%mod,bin[i]=(bin[i-]*G)%mod;
     Q=read();
     while(Q--)
     {
         int l=read(),r=read(),num=r-l+,ans=num;
         if (num==) write(),We;
         else
         {
             for (rg i=;i<ve[num].size();i++)
                 if (judge(ans/ve[num][i],l,r)) ans/=ve[num][i];
             write(ans),We;
         }
     }
 }