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

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

#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
='\0';///结束这个串
int ans=find(s,s1);
printf("%d\n",ans);
return 0;
}