bzoj3670:NOI2014 动物园

cnt[1](num1[1])=1是因为①边界的需要,根据结果推出来的 ②本身没有重叠的时候也是一种答案,num1[1]本来为0,最终答案会多1(除非j==0),所以直接把初始的变成1

因为在程序中i=1的时候不会计算,事实上所有next[i]=0的num1[i]=1

num1[0]=0不存在

ab找到的j=0

如果找到的j=0说明next=0,即根本不匹配,所以方案数为0,num1[0]=0

如果找到的j>0说明next>0,可以匹配,那么本身就是一种方案,再加上之前的可以抵消的,所以直接在i>=1且next[i]=0的num1[i]处赋值为1(本来是0在统计方案的时候会多那样的一种)

#include <istream>
#include <cstdio>
#include <cstring>
#include <cstdlib>
using namespace std;
#define L 1000010
#define P 1000000007
int n,m;
char s[L];
int next[L],num1[L];
long long ans;
void getnext()
{
    int j=0;next[1]=0;
    num1[1]=1;//!! num1即cnt数组
    for (int i=2;i<=m;i++)
    {
        while (j>0 && s[j+1]!=s[i]) j=next[j];
        if (s[j+1]==s[i]) j++;
        next[i]=j;
        num1[i]=num1[j]+1;//经过几次j变成0
    }
    j=0;next[0]=0;
    for (int i=2;i<=m;i++)
    {
        while (j>0 && s[j+1]!=s[i]) j=next[j];
        if (s[j+1]==s[i]) j++;
        while ((j<<1)>i) j=next[j];
        ans=(ans*(num1[j]+1)) % P;
    }
}
int main()
{
    scanf("%d", &n);
    while (n--)
    {
        scanf("%s", s+1);
        ans=1;
        m=strlen(s+1);
        getnext();
        printf("%lld\n", ans);
    }
}


你可能感兴趣的:(字符串,KMP,NOI)