signed

QiShunwang

“诚信为本、客户至上”

LeetCode 实现 strStr()

2021/6/24 15:20:01   来源:

题目:实现 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

解法

解法1:双层for,注意按照题目设定needle的size为0时返回0,而haystack的size0时返回-1。

第一层 for的条件i <= haystack.size() - needleSize

1. haystack.size() - needleSize,避免遍历过多的无用元素

2. 之所以是小于等于是因为有haystack与needle 的size大小相等的情侣需要考虑

class Solution {
public:
    int strStr(string haystack, string needle) {
        int needleSize = needle.size();
        int haystackSize = haystack.size();
        if (!needleSize)
        {
            return 0;
        }
        if (!haystackSize)
        {
            return -1;
        }
        if( needleSize > haystackSize)
        {
            return -1;
        }

        for(int i = 0; i <= haystack.size() - needleSize; i++)
        {
            int count = 0;
            for (int j = 0;j < needleSize;j++)
            {
                if (needle[j] == haystack[i+j])
                {
                    count++;
                }
                else
                {
                    break;
                }
            }
            if(count == needleSize)
            {
                return i;
            }
            
        }
        return -1;
    }
};

解法2:KMP算法

官方题解链接

https://leetcode-cn.com/problems/implement-strstr/solution/shi-xian-strstr-by-leetcode-solution-ds6y/