07 实现 strStr()（leecode 28）

1 问题

实现 strStr() 函数。

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

示例 1:
输入: haystack = “hello”, needle = “ll”
输出: 2

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

说明:
当 needle 是空字符串时，我们应当返回什么值呢？这是一个在面试中很好的问题。
对于本题而言，当 needle 是空字符串时我们应当返回 0 。这与C语言的 strstr() 以及 Java的 indexOf() 定义相符。

2 解法

采用KMP算法：
（1）根据needle字符串，求出needle的next数组。
（2）从文本串开头逐个匹配needle字符串中的字符，若遇到不匹配的字符，则根据next数组的value，退回needle字符串（value - 1）的索引位置重新匹配。
（3）本题要在文本串字符串中找出模式串出现的第一个位置 (从0开始)，所以返回当前在文本串匹配模式串的位置i 减去 模式串的长度，就是文本串字符串中出现模式串的第一个位置。（i已经在文本串中遍历到了完整的模式串，现要返回遍历的开始）

class Solution {
public:
    void getNext(int *next, string &s)
    {
        //指向前缀，同时也表示前后缀最大相等长度
        int j = 0;
        //next第一个值为0（一开始没有相等前后缀）
        next[0] = 0;
        for(int i = 1; i < s.size(); i++)
        {
            //1.前后缀不相等
            while(j > 0 && s[i] != s[j])
            {
                j = next[j - 1]; //回退到next数组的前一个下标
            }
            //2.前后缀相等
            if(s[i] == s[j])
            {
                j++;
            }
            next[i] = j;
        }
    }
    int strStr(string haystack, string needle) {
        //模式串为空，返回0
        if(needle.size() == 0)
        {
            return 0;
        }
        //定义next数组，其长度等于模式串的长度
        int next[needle.size()];
        //求取needle的next数组
        getNext(next, needle);
        int j = 0;
        //从头开始匹配
        for(int i = 0; i < haystack.size(); i++)
        {
            //1.前后缀不相等
            //模式串与文本串不等，则回退到next数组的前一个下标
            while(j > 0 && needle[j] != haystack[i])
            {
                j = next[j - 1];
            }
            //2.前后缀相等
            if(needle[j] == haystack[i])
            {
                j++;
            }
            //3.j运动到模式串尾部，说明匹配成功
            if(j == needle.size())
            {
                return (i - needle.size() + 1);
            }
        }
        //i移动至文本串末尾，而j未到模式串末尾，说明没有匹配的
        return -1;
    }
};