模板 KMP

  1 int next[N];
  2 char str1[M],str2[N];
  3 //str1 长，str2 短
  4 //len1,len2,对应str1,str2的长
  5 
  6 void get_next(int len2)
  7 {
  8     int i = 0,j = -1;
  9     next[0] = -1;
 10     while(i<len2)
 11     {
 12         if(j == -1 || str2[i] == str2[j])
 13         {
 14             i++;
 15             j++;
 16             if(str2[i] != str2[j])
 17                 next[i] = j;
 18             else
 19                 next[i] = next[j];
 20         }
 21         else
 22             j = next[j];
 23     }
 24     //计算某字符串的周期，如aaaa是4，abcd是1
 25     /*
 26     int i = 0;j = -1;
 27     next[0] = -1;
 28     while(str2[i])
 29     {
 30         if(j == -1 || str2[i] == str2[j])
 31         {
 32             i++;j++;
 33             next[i] = j;
 34         }
 35         else
 36         j = next[j];
 37     }
 38     len = strlen(str);
 39     i = len-j;
 40     if(len%i==0)
 41     return len/i;
 42     else
 43     return 1;
 44     */
 45 }
 46 
 47 int kmp(int len1,int len2)
 48 {
 49     int i = 0,j = 0;
 50     get_next(len2);
 51     while(i<len1)
 52     {
 53         if(j == -1 || str1[i] == str2[j])
 54         {
 55             i++;
 56             j++
 57         }
 58         else
 59             j = next[j];
 60         /*
 61                 if(j == len2)//计算str2在str1中出现多少次
 62                 {
 63         cnt++;
 64         j= next[j];
 65                 }
 66                 */
 67     }
 68     //return j; //j为匹配的长度
 69     if(j>len2)
 70         return 1;//这里也可以返回i-len2来获得匹配在主串中开始的位置
 71     else
 72         return 0;
 73 }
 74 
 75 //数字KMP
 76 int a[1000005],b[10005];
 77 int next[10005],n,m;
 78 
 79 void getnext()
 80 {
 81     int i = 0,j = -1;
 82     next[0] = -1;
 83     while(i<m)
 84     {
 85         if(j == -1 || b[i] == b[j])
 86         {
 87             i++;
 88             j++;
 89             if(b[i] == b[j])
 90             next[i] = next[j];
 91             else
 92             next[i] = j;
 93         }
 94         else
 95         j = next[j];
 96     }
 97 }
 98 
 99 int kmp()//返回匹配位置
100 {
101     int i = 0,j = 0;
102     while(i<n)
103     {
104         if(a[i] == b[j])
105         {
106             if(j == m-1)
107             return i-j+1;
108             i++;
109             j++;
110         }
111         else
112         {
113             j = next[j];
114             if(j == -1)
115             {
116                 i++;
117                 j = 0;
118             }
119         }
120     }
121     return -1;
122 }