题目
https://www.luogu.org/problemnew/show/P2375
思路
这题是KMP变式题NOI水题
一开始是想在next数组基础上判断(k<<1)>i+1的,但这会影响后面next数组求值。
正解是先求next数组,身边求num数组,然后再判断(k<<1)>i+1。
代码
#include<iostream>
#include<cstdio>
#include<cstring>
using namespace std;
const long long maxn=1e6+7,mod=1e9+7;
char st[maxn];
int next[maxn],num[maxn],n;
int main()
{
scanf("%d\n",&n);
while(n--)
{
memset(num,0,sizeof(num));memset(next,0,sizeof(next));
long long ass=1;
scanf("%s",st);
next[0]=-1;
int l=strlen(st),k=0;
for(int i=0; i<l; i++)
{
k=next[i];
while(k!=-1&&st[i]!=st[k]) k=next[k];
next[i+1]=++k;
num[i+1]=num[k]+1;
}
k=0;
for(int i=1; i<l; i++)
{
while(k!=-1&&st[i]!=st[k]) k=next[k];
k++;
while((k<<1)>i+1) k=next[k];
ass*=num[k]+1;ass%=(long long)mod;
}
cout<<ass<<endl;
}
}