Leetcode#28 Implement strStr()

原题地址

 

因为是道简单题，所以最简单的字符串匹配应该也能过，但还是练习一下KMP算法为好。KMP算法的介绍可以参考这篇

以后有时间试试Boyer-Moore算法

代码：

int strStr(char *haystack, char *needle) {
  if (!needle[0]) return 0;
  int n = strlen(needle);
  int *next = new int[n + 1];
  int i, j;

  i = 0;
  j = -1;
  next[0] = -1;
  while (needle[i]) {
    if (j < 0 || needle[i] == needle[j]) {
      i++;
      j++;
      next[i] = j;
    }
    else
      j = next[j];
  }

  i = 0;
  j = 0;
  while (haystack[i] && j < n) {
    if (j < 0 || haystack[i] == needle[j]) {
      i++;
      j++;
    }
    else
      j = next[j];
  }

  return j == n ? i - n : -1;
}