2795: [Poi2012]A Horrible Poem

2795: [Poi2012]A Horrible Poem

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

[ Submit][ Status][ Discuss]



假设当前询问的是子串[l,r]，长度为len，首先循环节长度a，要满足a|len

若a为当前子串的循环节，显然，a*t也是，那么，考虑逐步缩小答案

先筛好当前长度以内的所有素因子，对于每个询问先假定ans = len，

每次拿一个素因子k，测试ans / k是否满足条件

利用KMP算法的结论，子串[l,r]有长度为N的循环节的充要条件为子串[l,r-N]与[l+N,r]完全相同

筛法打残wa两次，，什么鬼。。。

#include<iostream>
#include<cstdio>
#include<cstring>
#include<vector>
#include<queue>
#include<algorithm>
#include<cmath>
using namespace std;

const int maxn = 5E5 + 5;
typedef unsigned long long UL;
const UL p = 233;

int n,m,tot,pri[maxn];
bool not_pri[maxn];
char s[maxn];
UL h[maxn],mi[maxn];

vector <int> v[maxn];

void Pre_Work()
{
	for (int i = 2; i <= n; i++)
	{
		if (!not_pri[i]) 
		{
			pri[++tot] = i;
			v[i].push_back(i);
		}
		else
		{
			int x = i;
			for (int j = 1; j <= tot; j++)
				if (x % pri[j] == 0)
				{
					v[i].push_back(pri[j]);
					while (x % pri[j] == 0) x /= pri[j];
					if (x == 1) break;
				}
		}
		for (int j = 1; j <= tot; j++)
		{
			int Nex = i*pri[j];
			if (Nex > n) break;
			not_pri[Nex] = 1;
			if (i % pri[j] == 0) break;
		}
	}
	mi[0] = 1;
	for (int i = 1; i <= n; i++)
		mi[i] = mi[i-1]*p;
}

UL Get(int l,int r)
{
	return h[r] - h[l-1]*mi[r-l+1];
}

bool Judge(int l,int r,int N)
{
	return Get(l,r - N) == Get(l + N,r);
}

int getint()
{
	char ch = getchar(); int ret = 0;
	while (ch < '0' || '9' < ch) ch = getchar();
	while ('0' <= ch && ch <= '9')
		ret = ret*10 + ch - '0',ch = getchar();
	return ret;
}

int main()
{
	#ifdef DMC
		freopen("DMC.txt","r",stdin);
	#endif
	
	n = getint(); scanf("%s",s + 1);
	for (int i = 1; i <= n; i++)
		h[i] = h[i-1]*p + s[i];
	m = getint(); Pre_Work();
	while (m--)
	{
		int l = getint(),r = getint();
		if (l == r) {puts("1"); continue;}
		int x = r - l + 1,Ans = x;
		for (int i = 0; i < v[x].size(); i++)
		{
			int y = v[x][i];
			while (Ans % y == 0 && Judge(l,r,Ans / y)) Ans /= y;
		}
		printf("%d\n",Ans);
	}
	return 0;
}