2017计蒜客第一场B题-阿里天池的新任务（简单）（数据结构-KMP）

本文链接：https://blog.csdn.net/lychee123/article/details/72588119

2017计蒜客第一场B题-阿里天池的新任务（简单）KMP子串匹配个数

题意

给出一定的规则找出长串s，已知短串s1，求s1在长串s中出现的次数
KMP

一个效率非常高的字符串匹配算法。
KMP是o（n+m）的算法

#include<bits/stdc++.h>
using namespace std;
typedef long long LL;
int n,a,b,l,r,i,j;
char s[1000010],s1[1000010];///s为长串,s1为短串
int f[1000010];
int w[1000010];
///KMP算法求最大匹配子串个数
void getf(char s1[])
{
    memset(f,0,sizeof(f));
    int len=strlen(s1);
    for(i=1;i<len;i++)
    {
        int j=f[i];
        while(j&&s1[i]!=s1[j])
            j=f[j];
        f[i+1]=(s1[i]==s1[j])?j+1:0;
    }
}
int find(char s[],char s1[])
{
    int ans=0;
    getf(s1);
    j=0;
    int len=strlen(s1);
    for(i=0;i<n;i++)
    {
        while(j&&s[i]!=s1[j])
            j=f[j];
        if(s[i]==s1[j])
            j++;
        if(j==len)
        {
            ans++;
        }
    }
    return ans;
}

int main()
{
    scanf("%d%d%d%d%d",&n,&a,&b,&l,&r);
    scanf("%s",s1);
    for(i=0;i<n;i++)
    {
        if(i==0) w[i]=b;
        else w[i]=(w[i-1]+a)%n;
        if(w[i]>=l&&w[i]<=r&&(w[i]%2==0))
            s[i]='A';
        if(w[i]>=l&&w[i]<=r&&w[i]%2==1)
            s[i]='T';
        if((w[i]<l||w[i]>r)&&w[i]%2==0)
            s[i]='G';
        if((w[i]<l||w[i]>r)&&w[i]%2==1)
            s[i]='C';
    }///按照规则得出这个s串
    s[n]='\0';///结束这个串
    int ans=find(s,s1);
    printf("%d\n",ans);
    return 0;
}