POJ 3461 kmp

今天重新读了一遍算法导论上的kmp算法，感觉理解比以前更加深刻了，随手写了这段代码。算法导论中，每次都是拿当前字符的下一个字符去比较，如果不能匹配成功的话，根据之前字符串的最长前缀和最长后缀进行移动字符串。注意注释的那一句，为了找到所有的字符串应该这样移动。

#include<stdio.h>
#include<stdlib.h>
#include<string.h>

char w[10005];
char t[1000005];
int next[10005];
int T;

void get_next(char str[],int len){
    int k=0;
    next[1]=0;
    for(int i=2;i<=len;i++){
	    while(k>0&&str[k+1]!=str[i])
	        k=next[k];
	    if(str[k+1]==str[i])
	        k++;
	    next[i]=k;
	}
}

int match(char P[],int lenp,char T[],int lent){
    int res=0;
	get_next(P,lenp);
    int k=0;
    for(int i=1;i<=lent;i++){
	    while(k>0&&P[k+1]!=T[i])
	        k=next[k];
	    if(P[k+1]==T[i])
	        k++;
	    if(k==lenp){ 
			res++;
			k=next[k];//这儿修改很重要，不然会超时 
		}
	}
	return res;
}

int main(){
	scanf("%d",&T);
	while(T--){
		memset(next,0,sizeof(next));
	    scanf("%s",w+1);
	    scanf("%s",t+1);
	    int lenw=strlen(w+1);
	    int lent=strlen(t+1);
	    int ans=match(w,lenw,t,lent);
	    printf("%d\n",ans);
	}
	//system("pause");
    return 0;
}