7

LeetCode-028-实现 strStr()

 2 years ago
source link: https://segmentfault.com/a/1190000040810335
Go to the source link to view the article. You can view the picture content, updated content and better typesetting reading experience. If the link is broken, please click the button below to view the snapshot at that time.

LeetCode-028-实现 strStr()

发布于 10 月 14 日

实现 strStr()

题目描述:实现 strStr() 函数。

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

示例说明请见LeetCode官网。

来源:力扣(LeetCode)
链接:https://leetcode-cn.com/probl...
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。

解法一:穷举法
  • 首先,如果needle为空,直接返回0;如果 haystack 为空 或者 haystack 的长度 小于 needle 的长度,直接返回-1;
  • 否则,从 haystack 的第一位开始跟 needle 进行匹配,如果匹配不上,则往后继续遍历haystack,直到遍历完成,就能得到结果。

说明:该方法效率比较差。

解法二:KMP算法

首先,构造一个next数组,先计算出下一次跳转的位置,然后遍历按照next数组的位置将原串和匹配串进行匹配。

import com.google.common.base.Strings;

public class LeetCode_028 {
    /**
     * 穷举法
     *
     * @param haystack
     * @param needle
     * @return
     */
    public static int strStr(String haystack, String needle) {
        if (needle == null || needle.length() == 0) {
            return 0;
        }
        if (haystack == null || haystack.length() == 0 || haystack.length() < needle.length()) {
            return -1;
        }
        int first = 0;
        while (first < haystack.length()) {
            int matchCount = 0;
            for (int i = 0; i < needle.length() && (i + first) < haystack.length(); i++) {
                if (needle.charAt(i) == haystack.charAt(i + first)) {
                    matchCount++;
                } else {
                    break;
                }
            }
            if (matchCount == needle.length()) {
                return first;
            } else {
                first++;
            }
        }
        return -1;
    }

    /**
     * KMP算法
     *
     * @param haystack 原串
     * @param needle   匹配串
     * @return
     */
    public static int strStr2(String haystack, String needle) {
        if (Strings.isNullOrEmpty(needle)) {
            return 0;
        }
        // 分别获取原串和匹配串的长度
        int haystackLength = haystack.length(), needleLength = needle.length();
        // 原串和匹配串前面都加一个空格,使其下标从1开始
        haystack = " " + haystack;
        needle = " " + needle;

        char[] haystackList = haystack.toCharArray();
        char[] needleList = needle.toCharArray();

        // 构建 next 数组,数组长度为匹配串的长度(next 数组是和匹配串相关的)
        int[] next = new int[needleLength + 1];
        // 构造过程 i = 2, j = 0 开始,i 小于等于匹配串长度【构造 i 从 2 开始】
        for (int i = 2, j = 0; i <= needleLength; i++) {
            // 匹配不成功的话,j = next[j]
            while (j > 0 && needleList[i] != needleList[j + 1]) {
                j = next[j];
            }
            // 匹配成功的话,先让 j++
            if (needleList[i] == needleList[j + 1]) {
                j++;
            }
            // 更新 next[i],结束本次循环, i++
            next[i] = j;
        }

        // 匹配过程,i = 1, j = 0 开始,i 小于等于原串长度【匹配 i 从 1 开始】
        for (int i = 1, j = 0; i <= haystackLength; i++) {
            // 匹配不成功 j = next[j]
            while (j > 0 && haystackList[i] != needleList[j + 1]) {
                j = next[j];
            }
            // 匹配成功的话,先让 j++,结束本次循环后 i++
            if (haystackList[i] == needleList[j + 1]) {
                j++;
            }
            // 整一段都匹配成功,直接返回下标
            if (j == needleLength) {
                return i - needleLength;
            }
        }

        return -1;
    }

    public static void main(String[] args) {
        System.out.println(strStr("mississippi", "issi"));

        System.out.println(strStr2("mississippi", "issi"));
    }
}

【每日寄语】 在最美的年华,做最喜欢的事情,别辜负了美好时光,借时光之手,暖一处花开,借一方晴空,拥抱梦想。


About Joyk


Aggregate valuable and interesting links.
Joyk means Joy of geeK