字符串 KMP HDU 2087

#include <stdio.h>
#include <string.h>
/*
abcde a3
aaaaaa  aa
#
水题KMP 
*/
const int M = 1005; 
const int N = 1005;
char a[N],b[M];
int next[M];
int lena,lenb;
void getNext()
{
	int j=-1;next[0]=-1;
	for(int i=1;i<lenb;i++)
	{
		while(j>=0 && b[i]!=b[j+1])
			j=next[j];
		if(b[i]==b[j+1])
			j++;
		next[i]=j;
	}
}
int kmp()
{
	int j=next[0];
	int ans=0;
	for(int i=0;i<lena;i++)
	{
		while(j>=0 && a[i]!=b[j+1])
			j=next[j];
		if(a[i]== b[j+1])
			j++;
		if(j==lenb-1)
		{
			ans++;j=next[0];//回到开头 
		}
	}
	return ans;
}
int main()
{
	while(~scanf("%s",a) && a[0]!='#')
	{
		scanf("%s",b);
		lena=strlen(a);lenb=strlen(b);
		getNext();
		printf("%d\n",kmp()); 
	}
	return 0;
}