POJ2406 Power Strings(Hash)

传送门

分析

枚举子串的长度，然后判断是否后面的没一段都与子串相等。
用字符串Hash判断是否相等。

代码

#include<bits/stdc++.h>
using namespace std;
#define INF 0x3f3f3f3f
#define full(a,b) memset(a,b,sizeof a)
#define ll long long
#define ui unsigned int
int read()//快读 
{
	int x=0,f=1;char ch=getchar();
	while(ch<'0'||ch>'9') f=ch=='-'?-1:1,ch=getchar();
	while(ch>='0'&&ch<='9') x=(x<<1)+(x<<3)+(ch^48),ch=getchar();
	return x*f;
}
#define debug 1
#define N 1000005
#define b 31
char arr[N];
ui Hash[N],p[N];
int main()
{
	if(debug==-1)
	{
		freopen("「POJ2406」Power Strings.in","r",stdin);
		freopen("「POJ2406」Power Strings.out","w",stdout);
	}
	p[0]=1;
	for(int i=1; i<N; i++)
		p[i]=p[i-1]*b;
	while(1)
	{
		scanf("\n%s",arr);
		if(arr[0]=='.') break;
		int len=strlen(arr);
		Hash[0]=arr[0];
		for(int i=1; i<len; i++)
			Hash[i]=Hash[i-1]*b+arr[i];//初始化 
		ui s=arr[0];//只需要用一个数来存Hash值 
		for(int i=0; i<len; i++)
		{
			if(i!=0) s=s*b+arr[i];//前缀累积子串的Hash值 ，避免循环 
			int flag=1,le=i+1;
			if(len%le) continue;//除不尽的一定不能是子串 
			for(int j=le; j+le-1<len; j+=le)
				if(s!=Hash[j+le-1]-Hash[j-1]*p[le])
					{flag=0;break;}//不相同就退出 
			if(flag)
			{
				printf("%d\n",len/le);//输出长度 
				break;//因为求最少长度，所以满足条件就退出
			}
		}
	}
	return 0;
}