leetcode 28. 实现 strStr() C++

leetcode 28. 实现 strStr() C++

难度：简单
给定一个 haystack 字符串和一个 needle 字符串，在 haystack 字符串中找出 needle 字符串出现的第一个位置 (从0开始)。如果不存在，则返回 -1。

示例 1:

输入: haystack = "hello", needle = "ll"
输出: 2

示例 2:

输入: haystack = "aaaaa", needle = "bba"
输出: -1

说明：当 needle 是空字符串时我们应当返回 0

思路：KMP，块内求出next[]数组，遍历一遍即可求解。

代码：

class Solution {
public:
    int strStr(string haystack, string needle) {
        if(!needle.size()) return 0;
        if(!haystack.size()) return -1;
        int m= haystack.size();
        int n=needle.size();
        
        //块内求解next[] 
        int next[n+1]; //int[110] 报边界错误，还是不能懒！
        {
        int j =-1;
        next[0]=-1;
        for(int i=1;i<n;i++){
            while(j!=-1 && needle[i]!=needle[j+1]){
                j=next[j];
            }
            if(needle[i]==needle[j+1]){
                j++;
            }
            next[i]=j;
        }
        }//

        //求解pos
        int j=-1;
        for(int i=0;i<m;i++){
            while(j!=-1 && haystack[i]!=needle[j+1]){
                j=next[j];
            }
            if(haystack[i]==needle[j+1]) 
                j++;
            if(j==n-1) return i-n+1; 
        }
        return -1;    
    }
};

复杂度分析：
若strlen(str)=m，strlen(substr)=n，则：

• 空间复杂度：$O(n)$，使用一个next[]数组；
• 时间复杂度：$O(m+n)$；