28. 实现 strStr()

28. 实现 strStr()

实现 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 仅由小写英文字符组成

class Solution {
    public int strStr(String haystack, String needle) {
        if(needle.length() == 0){
            return 0;
        }
        // 根据模式串去找出对应的next数组
        int next[] = new int[needle.length()];
        // 暂存的数据变量，为了描述方便，我们如下假定i为needle的循环下标，
        // now是next数组中从[i - now,i]这一串子串与开头前缀字符匹配的最大长度。也可以看作是前缀已经匹配到了第几个字符。
        // 即我们每次都是比较needle[i]和next[now]
        //      如果相同，说明最长串又长了
        //      如果不同，说明最长串已经戛然而止，我们需要从next[i-now,now]中找到以next[now - 1]为结尾的最长匹配字符。
        //                而next[now - 1]这个本身就已经在前缀中出现了，我们只要查询next[next[now - 1]]就能知道next[now - 1]是第几个。
        //                同时要令i减1，因为此时比较的now已经变了。即刚刚轮循环应该视为白费。
        int now = 0;
        for(int i = 1; i < needle.length(); ++i){
            if(needle.charAt(i) == needle.charAt(now)){
                ++now;
                next[i] = now;
            } else {
                if(now > 0) {
                    now = next[now - 1];
                    --i;
                }
            }
        }
        for(int i = 0, j = 0; i < haystack.length(); ++i){
            // 如果一样，自然是++j
            if(haystack.charAt(i) == needle.charAt(j)){
                ++j;
            } else {
                // 否则的话，就要看j已经是第几个字符，这个数量拿到next数组中，就知道该字符可以视为从哪里开始的新最长匹配字符串。同理j要减1.
                if(j != 0) {
                    j = next[j - 1];
                    --i;
                }
            }
            if(j == needle.length()){
                return i - j + 1;
            }
        }
        return -1;
    }
}