leetcode算法每天一题028:找出字符串中第一个匹配项的下标

题目描述

给你两个字符串 haystack 和 needle ，请你在 haystack 字符串中找出 needle 字符串的第一个匹配项的下标（下标从 0 开始）。如果 needle 不是 haystack 的一部分，则返回 -1 。

输入：haystack = "sadbutsad", needle = "sad"
输出：0
解释："sad" 在下标 0 和 6 处匹配。
第一个匹配项的下标是 0 ，所以返回 0 。


class Solution {
public:
    int strStr(string haystack, string needle) {

    }
};

解：

class Solution {
public:
    int strStr(string s, string p) {
      int n = s.length(),m = p.length();
      if(m==0) return 0;

      s.insert(s.begin(),' ');
      p.insert(p.begin(),' ');

      vector<int> next(m+1);

      for (int i=2,j=0;i<=m;i++){
          while(j>0&&p[i]!=p[j+1]) j =next[j];
          if(p[i] == p[j+1]) j++;
          next[i] = j;
      }

      for(int i=1,j=0;i<=n;i++){
          while(j>0 && s[i]!= p[j+1]) j= next[j];
          if(s[i]== p[j+1]) j++;
          if(j==m) return i-m;
      }  
      return -1;

    }
};