bzoj 3670 动物园 KMP

来源：互联网发布：自身投资知乎编辑：程序博客网时间：2024/05/21 09:46

KMP非常神，我们可以顺着fail数组把一个字符串所有前缀等于后缀的子串跑一边，而且顺序是从大到小。例如ababcccabab，fail11=4，fail4=2，fail2=0；对于这道题来说，我们可以顺着每一位的fail跑一遍（当然会T）求num数组，但是我们只要求出跑了多少次后，fail值等于0就可以了，上例sum[11]=sum[4]+1，sum在初始化的时候求出来。统计答案的时候，num[i]=sum[j]，j是顺着fail数组跑，长度<=(i/2)的第一个值；

#include<iostream>#include<cstdio>#include<cstring>#define LL long longusing namespace std;char s2[1000005];LL fail[1000005];LL  sum[1000005];LL  num[1000005];void init(int m){    fail[0]=0;fail[1]=0;    sum[0]=0;sum[1]=1;    for(int i=2;i<=m;i++)    {        int p=fail[i-1];        while(p&&s2[i]!=s2[p+1])p=fail[p];        if(s2[p+1]==s2[i]) p++;        fail[i]=p;        sum[i]=sum[p]+1;    }}void get(int m){    int p=0;num[0]=0;    for(int i=2;i<=m;i++)    {        while(p&&s2[i]!=s2[p+1])p=fail[p];        if(s2[i]==s2[p+1]) p++;        while(p>i/2) p=fail[p];        num[i]=sum[p];    }}LL ans=1,k=1e9+7;int main(){    int n;scanf("%d",&n);    while(n--)    {        ans=1;        scanf("%s",s2+1);        int L=strlen(s2+1);        init(L);get(L);        for(int i=1;i<=L;i++)        {            ans*=(num[i]+1);            ans%=k;        }        printf("%lld\n",ans);    }}

6 0