Implement strstr()

Implement strStr().

Returns the index of the first occurrence of needle in haystack, or -1 if needle is not part of haystack.

class Solution {
public: 
    void getNext(vector<int> &next, string &needle) {
          int i = 0, j = -1;
          next[i] = j;
          while (i != needle.length()) {
              while (j != -1 && needle[i] != needle[j]) j = next[j];
              next[++i] = ++j;
          }
     }
     int strStr(string haystack, string needle) {
         if (haystack.empty()) return needle.empty() ? 0 : -1;
         if (needle.empty()) return 0;
         vector<int> next(needle.length() + 1);
         getNext(next, needle);
         int i = 0, j = 0;
         while (i != haystack.length()) {
             while (j != -1 && haystack[i] != needle[j]) j = next[j];
             ++i; ++j;
             if (j == needle.length()) return i - j;
         }
         return -1;
     }
};


你可能感兴趣的:(C++)