BZOJ2795/POI2012 A horrible poem

Task
給出一個由小寫英文字母組成的字符串S,再給出q個詢問,要求回答S某個子串的最短循環節。
如果字符串B是字符串A的循環節,那麼A可以由B重複若干次得到。
n<=500,000, q<=2,000,000.

Solution
假如最短循環節爲len ,設t=n/len ,現在的問題就是找到最大的t.
t肯定是n的因子,那就可以直接枚舉n的所有素因子k,若當前的串可以分成k份,那麼把n/k,把串變爲原來的1/k,繼續操作,直到枚舉所有的素因子.
如何判斷一個串[l,r] 是否可以分成相同的k份?哈希!!
直接比較[l,rk][l+k,r] 這兩段是否完全相同即可.
若相等:
保證[l,k1]=[l+k,l+2k1]=[l+3k,l+4k1]=[rk,r] ,滿足了循環的性質.
一個數最多有logn 個因子,最後複雜度爲O(qlogn) .

#include<cstdio>
#include<cstring>
#include<iostream>
#include<algorithm>
#include<vector>
#define ll long long
#define ull unsigned long long 
using namespace std;
inline void rd(int &res){
    res=0;char c;
    while(c=getchar(),c<48);
    do res=(res<<1)+(res<<3)+(c^48);
    while(c=getchar(),c>=48);
}
inline void print(int x){
    if(!x)return ;
    print(x/10);
    putchar((x%10)^48);
}
inline void sc(int x){
    if(x<0){x=-x;putchar('-');}
    print(x);
    if(!x)putchar('0');
    putchar('\n');
}
const int M=500005;
const int P=1e9+7,Lay=233;
int n,mark[M],son[M][10];
char A[M];
ull v[M],p[M];
void Init(){
    for(int i=2;i*i<M;i++){
        if(!mark[i]){
            for(int j=i;j<M;j+=i){
                son[j][++son[j][0]]=i;
                mark[j]=1;
            }
        }
    }
}
int get(int l,int r){
    return v[r]-v[l-1]*p[r-l+1];
}
bool chk(int st,int en,int len){//以st爲起點,len爲循環節 
    if(st==en&&len==1)return true;
    return get(st,en-len)==get(st+len,en);
}
int main(){
    int i,j,k,a,b,Q;
    scanf("%d",&n);
    scanf("%s",A+1);
    p[0]=1;
    for(i=1;i<=n;i++){
        v[i]=v[i-1]*Lay+A[i];
        p[i]=p[i-1]*Lay;
    }
    Init();
    scanf("%d",&Q);
    for(i=1;i<=Q;i++){
        rd(a);rd(b);
        int x=b-a+1,y=b-a+1,z=b-a+1;
        for(j=1;j<=son[x][0];j++){
            while(y%son[x][j]==0){
                z/=son[x][j];
                if(chk(a,a+y-1,y/son[x][j]))y/=son[x][j];
                else break;
            }
            while(z%son[x][j]==0)z/=son[x][j];
        }
        if(z!=1)if(chk(a,a+y-1,y/z))y/=z;
        sc(y);
    }
    return 0;
}

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