KMP算法

Posted 2020-10-31 jaydenouyang

为了解决字符串匹配问题

给你两个字符串，寻找其中一个字符串是否包含另一个字符串，如果包含，返回包含的起始位置。

 1 //str为母串
 2 //prt为子串
 3 
 4 
 5 void cal_next(char *str, int *next, int len)
 6 {
 7     next[0] = -1;//next[0]初始化为-1，-1表示不存在相同的最大前缀和最大后缀
 8     int k = -1;//k初始化为-1
 9     for (int q = 1; q <= len-1; q++)
10     {
11         while (k > -1 && str[k + 1] != str[q])//如果下一个不同，那么k就变成next[k]，注意next[k]是小于k的，无论k取任何值。
12         {
13             k = next[k];//往前回溯
14         }
15         if (str[k + 1] == str[q])//如果相同，k++
16         {
17             k = k + 1;
18         }
19         next[q] = k;//这个是把算的k的值（就是相同的最大前缀和最大后缀长）赋给next[q]
20     }
21 }
22 
23 int KMP(char *str, int slen, char *ptr, int plen)
24 {
25     int *next = new int[plen];
26     cal_next(ptr, next, plen);//计算next数组
27     int k = -1;
28     for (int i = 0; i < slen; i++)
29     {
30         while (k >-1&& ptr[k + 1] != str[i])//ptr和str不匹配，且k>-1（表示ptr和str有部分匹配）
31             k = next[k];//往前回溯
32         if (ptr[k + 1] == str[i])
33             k = k + 1;
34         if (k == plen-1)//说明k移动到ptr的最末端
35         {
36             //cout << "在位置" << i-plen+1<< endl;
37             //k = -1;//重新初始化，寻找下一个
38             //i = i - plen + 1;//i定位到该位置，外层for循环i++可以继续找下一个（这里默认存在两个匹配字符串可以部分重叠），感谢评论中同学指出错误。
39             return i-plen+1;//返回相应的位置
40         }
41     }
42     return -1;  
43 }