【jzoj4624】【字符串匹配】【kmp】

题目大意

求一个串在另一个串复制n遍的串出现几遍。

题解

显然将两个串做kmp，有很多重复，可以先把第二个串复制到比第一个串长，计算结果，增长一个串，计算结果，计算有多少个增长串，得到最终结果。

code

#include<set>
#include<cmath>
#include<cstdio>
#include<cstring>
#include<algorithm>
#define LL long long
#define fo(i,j,k) for(int i=j;i<=k;i++)
#define fd(i,j,k) for(int i=j;i>=k;i--)
using namespace std;
LL const maxn=100000;
LL n,sn,tn,fail[maxn+10],match[maxn*5+10];
char s[maxn+10],t[maxn*5+10];
int main(){
    //freopen("len.in","r",stdin);
    //freopen("len.out","w",stdout);
    freopen("d.in","r",stdin);
    freopen("d.out","w",stdout);
    scanf("%lld\n",&n);
    scanf("%s\n",s+1);sn=strlen(s+1);
    scanf("%s\n",t+1);tn=strlen(t+1);
    LL m=(sn-1)/tn+1;
    fo(i,tn+1,tn*m+tn)t[i]=t[i-tn];
    fo(i,2,sn){
        fail[i]=fail[i-1];
        for(;fail[i]&&(s[fail[i]+1]!=s[i]);fail[i]=fail[fail[i]]);
        if(s[fail[i]+1]==s[i])fail[i]++;
    }
    fo(i,1,min(tn*n,tn*m+tn)){
        match[i]=match[i-1];
        for(;match[i]&&(s[match[i]+1]!=t[i]);match[i]=fail[match[i]]);
        if(s[match[i]+1]==t[i])match[i]++;
    }
    if(tn*n<=tn*m+tn){
        LL ans=0;
        fo(i,1,tn*n)if(match[i]==sn)ans++;
        printf("%lld",ans);
        return 0;
    }
    LL tmp=0;
    fo(i,1,tn*m)if(match[i]==sn)tmp++;
    LL tmp2=0;
    fo(i,tn*m+1,tn*m+tn)if(match[i]==sn)tmp2++;
    printf("%lld",tmp+tmp2*(n-m));
    return 0;
}