LeetCode28—实现strStr()（java版）

题目描述：

标签：字符串   双指针

实现 strStr() 函数。

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

 代码：

思路分析：这里采用了kmp算法来解决这道问题。（顺便复习了一下这个算法的全过程）

1、首先要定义一个方法kmpNext（子字符串），获取子字符串的部分匹配值表。

①首先定义一个next[子字符串长度]数组，保存各个位置的部分匹配值，当长度为1是值为0，所以next[0]=0;

②定义双指针i,j，j=0，让i=1开始向后移动遍历，当s[i]==s[j]时，j向后移动j++。同时使next[i]=j。（在这中间当出现j>0时，要比较当前s[i]和s[j]是否不等，如果不等则将j回溯即j=next[j-1]）

③得到next[]部分匹配值表返回

2、然后定义一个kmp搜索算法kmpSearch(s1,s2,next)，获取索引。（其实思路和查找kmpNext很相似，不过从参数从一个字符串变成了两个）

①定义双指针i,j，i=0指向字符串s1,j=0指向字符串s2，让i=0开始向后移动遍历，当s[i]==s[j]时，j向后移动j++，在这中间当出现j>0时，要比较当前s[i]和s[j]是否不等，如果不等则将j回溯即j=next[j-1]

②当出现j=s2.length()，说明已经匹配上所有的字符，返回索引i-j+1

③如果没有匹配成功，则返回0

3、不过根据题目提示的“当 needle 是空字符串时我们应当返回 0 ”，所以需在查找前先判断needle.length()==0，如果true则返回0。（不加这个判断有个测试案例过不去！）

class Solution {
    public int strStr(String haystack, String needle) {
        if(needle.length() == 0){
            return 0;
        }
        int[] next = kmpNext(needle);
        int ans = kmpSearch(haystack,needle,next);
        return ans;
    }

    //定义方法获取next数组即部分匹配值表
    public int[] kmpNext(String s){
        int[] next = new int[s.length()];
        next[0] = 0;//字符串长度为1是部分匹配值是0
        for(int i = 1,j = 0;i < s.length();i++){
            while(j > 0 && s.charAt(i) != s.charAt(j)){
                j = next[j - 1];
            }

            if(s.charAt(i) == s.charAt(j)){
                j++;
            }
            next[i] = j;
        }
        return next;
    }

    //定义kmp搜索算法
    public int kmpSearch(String s1,String s2,int[] next){
        for(int i = 0,j = 0;i < s1.length();i++){
            while(j > 0 && s1.charAt(i) != s2.charAt(j)){
                j = next[j - 1];
            }

            if(s1.charAt(i) == s2.charAt(j)){
                j++;
            }

            if(j == s2.length()){
                return (i - j + 1);
            }
        }
        return -1;
    }
}