28. 实现 strStr()

难度简单832
实现 strStr() 函数。
给你两个字符串 haystack和needle,请你在haystack字符串中找出needle字符串出现的第一个位置(下标从 0 开始)。如果不存在,则返回 -1。
说明:
当 needle 是空字符串时,我们应当返回什么值呢?这是一个在面试中很好的问题。
对于本题而言,当 needle 是空字符串时我们应当返回 0 。这与 C 语言的 strstr() 以及 Java 的 indexOf()) 定义相符。
示例 1:
输入:haystack = “hello”, needle = “ll”
输出:2

示例 2:
输入:haystack = “aaaaa”, needle = “bba”
输出:-1

示例 3:
输入:haystack = “”, needle = “”
输出:0

提示:

  • 0 <= haystack.length, needle.length <= 5 * 104
  • haystack和needle仅由小写英文字符组成
    1. class Solution {
    2. public:
    3. int strStr(string haystack, string needle) {
    4. int len1 = haystack.size(), len2 = needle.size();
    5. if(!len2) return 0;
    6. vector<int> next(len2);
    7. //预处理next
    8. for(int i = 1, j = 0; i < len2; ++ i){
    9. while(j > 0 && needle[i] != needle[j]){
    10. j = next[j - 1];
    11. }
    12. if(needle[i] == needle[j]){
    13. j ++;
    14. }
    15. next[i] = j;
    16. }
    17. //进行匹配
    18. for(int i = 0, j = 0; i < len1; ++ i){
    19. while(j > 0 && haystack[i] != needle[j]){
    20. j = next[j - 1];
    21. }
    22. if(haystack[i] == needle[j]){
    23. j ++;
    24. }
    25. if(j == len2){
    26. return i - len2 + 1;
    27. }
    28. }
    29. return -1;
    30. }
    31. };