[NOI2014]动物园

题目

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;
    }
}