bzoj3670: [Noi2014]動物園(kmp)

Hanks_o發表於2018-04-18

題目傳送門

解法:
感覺上是kmp。
然後碼了個kmp。
然後看成了num[i]是字首字尾不相交的最長長度。
然後樣例出來個144。
發現並沒有什麼問題。
然後發現看錯題了。。

然後我就在此基礎上繼續做。
s[i]表示字首等於字尾且可以相交有多少個串。
那麼求出最長不相交長度。這個的s就是num[i]了。

程式碼實現:

#include<cstdio>
#include<cstring>
#include<cstdlib>
#include<iostream>
#include<algorithm>
#include<cmath>
#include<queue>
using namespace std;
typedef long long ll;
int p[1100000],next[1100000],s[1100000];
char ss[1100000];const ll mod=1000000007;
int main() {
    int T;scanf("%d",&T);
    while(T--) {
        scanf("%s",ss+1);int len=strlen(ss+1);
        p[1]=0;s[1]=1;
        for(int i=2;i<=len;i++) {
            int j=p[i-1];
            while(ss[j+1]!=ss[i]&&j>0)j=p[j];
            if(ss[j+1]==ss[i])j++;p[i]=j;
            s[i]=s[j]+1;
        }
        next[1]=0;
        for(int i=2;i<=len;i++) {
            if(p[i]<=i/2)next[i]=p[i];
            else {
                int j=next[i-1];
                if(j>0&&(ss[j+1]!=ss[i]||(j+1)>i/2))j=p[j];
                if(ss[j+1]==ss[i])j++;next[i]=j;
            }
        }ll ans=1;
        for(int i=1;i<=len;i++)ans=ans*ll(s[next[i]]+1)%mod;
        printf("%lld\n",ans);
    }
    return 0;
}

相關文章