牛客 白兔的字符串（哈希，二分）

链接：https://ac.nowcoder.com/acm/problem/15253
来源：牛客网

题目描述

白兔有一个字符串T。白云有若干个字符串S1,S2…Sn。

白兔想知道，对于白云的每一个字符串，它有多少个子串是和T循环同构的。

提示：对于一个字符串a，每次把a的第一个字符移动到最后一个，如果操作若干次后能够得到字符串b，则a和b循环同构。

所有字符都是小写英文字母

输入描述:

第一行一个字符串T（|T|<=10^6）
第二行一个正整数n (n<=1000)
接下来n行为S1~Sn (|S1|+|S2|+…+|Sn|<=10^{7)，max(|S1|,|S2|,|S3|,|S4|,…|Sn|)<=10}6

输出描述:

输出n行表示每个串的答案
示例1
输入
复制
abab
2
abababab
ababcbaba
输出
复制
5
2

分析：

将字符串T增加一倍，计算哈希值，则可得到T所有循环同构体的哈希值，并且排序。对字符串si，计算哈希值，然后二分查询是否存在同分异构。

代码

#include<bits/stdc++.h>
#define ll long long
using namespace std;
typedef unsigned long long ull;
const int N = 2e6+10;
char a[N],s[N];
ull p,h[N],q[N],hs[N];
int main()
{
	scanf("%s",a);
	int n=strlen(a);
	p=1;
	for(int i=1;i<=n;i++)
	{
		p=p*131;
	}
	for(int i=n;i<2*n;i++)
	{
		a[i]=a[i-n];
	}
	h[0]=a[0]-'0';
	for(int i=1;i<2*n;i++)
	{
		h[i]=h[i-1]*131+a[i]-'0';
	}
	int k=1;
	q[1]=h[n-1];
	for(int i=n;i<2*n;i++)
	{
		q[++k]=h[i]-h[i-n]*p;
	}
	sort(q+1,q+1+k);
	int T;
	scanf("%d",&T);
	while(T--)
	{
		scanf("%s",s);
		int m=strlen(s);
		hs[0]=s[0]-'0';
		for(int i=1;i<m;i++)
		{
			hs[i]=hs[i-1]*131+s[i]-'0';
		}
		ull sum,ans=0;
		for(int i=n-1;i<m;i++)
		{
			if(i==n-1) sum=hs[n-1];
			else sum=hs[i]-hs[i-n]*p;
			int x=lower_bound(q+1,q+1+k,sum)-q;
			if(q[x]==sum) ans++;
		}
		printf("%llu\n",ans);
	}
	return 0;
 }