扩展KMP的模版

1. #include <stdio.h>  
2. #include <stdlib.h>  
3. #include <string.h>  
4. #include <iostream>  
5. #include <algorithm>  
6. using namespace std;  
7. const int maxn=100005;  
8. int next[maxn],extand[maxn];  
9. char S[maxn],T[maxn];  
10. void GetNext(const char *T){  
11.      int len=strlen(T),a=0;  
12.      next[0]=len;  
13.      while(a<len-1 && T[a]==T[a+1]) a++;  
14.      next[1]=a;  
15.      a=1;  
16.      for(int k=2;k<len;k++){  
17.          int p=a+next[a]-1,L=next[k-a];  
18.          if( (k-1)+L >= p){  
19.              int j = (p-k+1)>0 ? (p-k+1) : 0;  
20.              while(k+j<len && T[k+j]==T[j]) j++;  
21.              next[k]=j;  
22.              a=k;  
23.          }  
24.          else  
25.              next[k]=L;  
26.      }  
27. }  
28. void GetExtand(const char *S,const char *T){  
29.      GetNext(T);  
30.      int slen=strlen(S),tlen=strlen(T),a=0;  
31.      int MinLen = slen < tlen ? slen : tlen;  
32.      while(a<MinLen && S[a]==T[a]) a++;  
33.      extand[0]=a;  
34.      a=0;  
35.      for(int k=1;k<slen;k++){  
36.          int p=a+extand[a]-1, L=next[k-a];  
37.          if( (k-1)+L >= p){  
38.              int j= (p-k+1) > 0 ? (p-k+1) : 0;  
39.              while(k+j<slen && j<tlen && S[k+j]==T[j]) j++;  
40.              extand[k]=j;  
41.              a=k;  
42.          }  
43.          else  
44.              extand[k]=L;  
45.      }  
46. }  
47. int main(){  
48.     while(scanf("%s%s",S,T)!=-1){  
49.          GetExtand(S,T);  
50.          for(int i=0;i<strlen(T);i++)  
51.              printf("%d ",next[i]);  
52.          puts("");  
53.          for(int i=0;i<strlen(S);i++)  
54.              printf("%d ",extand[i]);  
55.          puts("");  
56.     }  
57.     return 0;  
58. }  

这篇文章看的非常好，不懂得可以看这里 点击打开链接