哥也能写KMP了——实现strstr()

来源：互联网发布：bpm节拍软件编辑：程序博客网时间：2024/05/02 00:19

经过上次去面试，面试官要求实现strstr()，当场就蒙了。这个题目是模式匹配问题，《算法导论》里列出了几种字符串匹配算法：

朴素算法 | Rabin-Karp | 有限自动机算法 | Knuth-Morris-Pratt (KMP)

各种方法都有自己的优缺点，我觉得，还有一些方法可以参考：

1）比如像求最长公共子串那样，用动态规划，最后判断最长公共子串的最大值是不是模式串的长度，不过，这有点绕。

2）用后缀数组，这个也有点绕，实现起来也有点麻烦，就不说了。

个人觉得，还是KMP好，KMP该怎么写呢，立马抄起书《数据结构（C语言版）》，看了一下，感觉，KMP实现起来，代码是很少的，效率还算可以了，实现的过程中，难就难在如何构造next[] 数组，以及如何理解next[]，对避免指针无效回退的作用。看了一个上午，总算是明白了。现就贴下代码，做下笔记。

#include<iostream>using namespace std;void get_next(const char* s, int* next){  int i = 0, j = -1;  next[0] = -1;  int len = strlen(s);  while(i < len){    if(j == -1 || s[i] == s[j]){      ++i, ++j;      if(s[i] != s[j]) next[i] = j;      else next[i] = next[j];    }    else j = next[j];  }}int KMP_strstr(const char* s, const char* p){  int i = 0, j = 0;  int len_s = strlen(s), len_p = strlen(p);  int* next = new int[len_p];  get_next(s, next);  while(i < len_s && j < len_p){    if(j == -1 || s[i] == p[j])      ++i, ++j;    else j = next[j];  }  delete []next;  if(j == len_p) return i - len_p;  else return -1;}int main(){  const int len = 6;  const char *s = "abaabc";  cout<<KMP_strstr("aaabbabaabcaad", s);  return 0;}//output: 5

1 0