Codeforces 625B【KMP】

#include <bits/stdc++.h>
using namespace std;
typedef long long LL;
const int MAX=1000010;
int n,m;
char a[MAX],b[MAX];
int nextval[MAX];
LL ans;
void get_next(char b[],int m)
{
int i=0;
nextval[0]=-1;
int j=-1;
while(i<m)
{
if(j==-1||b[i]==b[j])
{
++i;
++j;
if(b[i]==b[j])
nextval[i]=nextval[j];
else
nextval[i]=j;
}
else
j=nextval[j];
}
}
void KMP(char a[],int n,char b[],int m,int nextval[],int pos)
{
int i=pos;
int j=0;
while(i<n)
{
if(j==-1||a[i]==b[j])
{
i++;
j++;
if(j==m)	//修改处
{
j=0;
ans++;
}
}
else
j=nextval[j];
}
}
int main()
{
int t,i,j;
while(~scanf("%s%s",a,b))
{
int p;
ans=0;

n=strlen(a);
m=strlen(b);
get_next(b,m);
KMP(a,n,b,m,nextval,0);
printf("%lld\n",ans);
}
return 0;
}