字符串实现strStr()——查找子串位置

Posted 2021-08-06 AI 菌

一、题目

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

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

二、题解

解析1： 对于本题，最简单的方法是使用字符串成员函数find()，可直接获取字串最先出现的位置。

解析2： 如果面试官不允许使用内置函数find()来实现，则需要自己实现字符串匹配，具体做法如下：

• 从左往右遍历字符串，寻找第一个字符相等的位置
• 若第一个字符相等，则依次比较后面的字符串。
• 若后面所有的字符都相等，则可直接返回之前记录的位置。
• 否则，继续遍历字符串，寻找下一个第一个字符相等的位置。
• 直到遍历结束，如果没有返回位置，则说明没有匹配成功，此时直接返回-1。

三、题解

方法1：

class Solution {
public:
    int strStr(string haystack, string needle) {
        int pos = haystack.find(needle);
        if(pos != string::npos)
            return pos;
        else
            return -1;
    }
};

方法2：

class Solution {
public:
    int strStr(string haystack, string needle) {
        if(needle=="") return 0;
        int n = haystack.size(), m = needle.size();

        for(int i=0; i<=n-m; ++i)  //从左至右遍历字符串
        {
            if(haystack[i]==needle[0]) //当needle第一个字符串相同时，进一步判断
            {
                int j = i+1;   //定位到下一个字符  
                int k = 1;     //定位到下一个字符
                while(j<n && k<m && haystack[j]==needle[k]) //判断是否继续相等
                {
                    j++;
                    k++;
                }
                if(k==m)  //遍历完退出while后，当k=m，则匹配成功，返回位置i；
                {
                    return i;
                }         
            }
        }
        return -1; //若字符串遍历完，没有匹配成功，则直接返回-1。
    }
};