bzoj3670 [Noi2014]动物园 kmp

bzoj3670 [Noi2014]动物园 kmp
题意:长度<=100w的字符串 num[i]表示前i个字符组成的字符串中既是前缀又是后缀且前缀后缀不重叠的子字符串个数 求这里写图片描述%(1e9+7)
题解:开始看错题了……以为num[i]是既是前缀又是后缀且前缀后缀不重叠的最长子字符串长度……翻题解没弄懂他们的方法……自己yy一个没敢写呢仔细一看题卧槽……这个事情告诉我们仔细看样例啊……
num[i]代表长度为i的前缀经过几次j=next[j]会得到0,然后重新匹配一次,这次注意当j*2>i的时候令j=next[j]即可

#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<iostream>
#include<algorithm>
#define ll long long
using namespace std;
const int Mo=(int)1e9+7;
const int N=1000100;
char a[N];
int next[N],num[N];
void kmp(){
    memset(num,0,sizeof(num));
    memset(next,0,sizeof(next));
    int j;
    scanf("%s\n",a+1);
    next[1]=0;j=0;num[1]=1;
    for(int i=2;a[i];i++){
        while(j && a[i]!=a[j+1]) j=next[j];
        j+=(a[i]==a[j+1]);
        next[i]=j;
        num[i]=num[j]+1;
    }
    //for(int i=1;a[i];i++) printf("%d ",next[i]);
    j=0;ll ans=1;
    for(int i=1;a[i];i++){
        while(j && a[i]!=a[j+1]) j=next[j];
        j+=(a[i]==a[j+1]);
        while(j*2>i) j=next[j];
        ans=ans*(num[j]+1);ans%=Mo;
    }
    printf("%lld\n",ans);
}
int main(){
    freopen("in.txt","r",stdin);
    freopen("out.txt","w",stdout);
    int T;
    scanf("%d\n",&T);
    while(T--) kmp();
    return 0;
}

你可能感兴趣的:(KMP,NOI,bzoj)