力扣（LeetCode）试题28-实现strStr() C++代码

没想到的点：若haystack="mississippi"，needle = "issip"。按我的匹配算法（在haystack中检索needle的第一个元素，若没有该元素返回-1，若有则搜索后边的元素是否对应，这是有问题的！！），匹配到haystack中前部分issis，程序就结束了，返回了-1，实际应该匹配到issip。缜密的逻辑还需要不断提升呀！（基于此思想改动的代码已经补上，在最后：即在haystack中，第一个i不满足条件，循环退出，找寻第二个i，第三个i，直到满足条件，否则返回-1）

错误代码记录一下吧，如下：

class Solution
{
public:
    int strStr(string haystack, string needle)
    {
        if (needle.length() == 0) return 0;
        else {
            int i = 0;
            while (i < haystack.length())
            {
                if (haystack[i] != needle[0])
                {
                    i += 1;
                }
                else
                    //否则，haystack中存在needle中的第一个元素，然后判断是不是needle整个元素
                {
                    int k = 0; //指向needle的头元素
                    int limit = needle.length();
                    while (k < limit)
                    {
                        if (haystack[i] != needle[k])
                            return -1;
                        else
                        {
                            i += 1;
                            k += 1;
                        }
                    }
                    return i - k;//循环完毕，此时i指向了needle最后一个元素，需要减去这个needle的长度
                }
            }
            return -1;
        }
    }

};

使用substr()方法的代码如下：

#include <iostream>

using namespace std;

class Solution
{
public:
    int strStr(string haystack, string needle)
    {
        int needle_size = needle.length();
        int haystack_size = haystack.length();
        int res = haystack_size - needle_size;
        if (needle_size == 0) return 0; //特殊情况
        else 
        {
            //遍历haystack，寻找是否包含needle
            for (int i = 0; i <= res; ++i)
            {
                string str = haystack.substr(i, needle_size);//substr从第i个位置截取，截取的长度为needle_size
                if (str == needle) return i;
                else continue;
            }
            return -1;
        }
    }

};

int main()
{
    string haystack = "mississippi";
    string needle = "issip";
    Solution sol;
    int p;
    p = sol.strStr(haystack, needle);
    cout << p << endl;
    int u;
    cin >> u;
    return 0;
}

 

修改正确的代码如下：（啊，头秃）

#include <iostream>

using namespace std;

class Solution
{
public:
    int strStr(string haystack, string needle)
    {
        int haystack_size = haystack.length();
        int needle_size = needle.length();
        int res = haystack_size - needle_size;

        if (needle_size == 0) return 0;//特殊情况

        else {
            int i = 0;//指针，指向haystack中的元素
            while (i <= res) //外层循环，用于找needle头元素在haystack中的位置。
            {
                if (haystack[i] != needle[0])
                {
                    i += 1;
                }
                else
                    //在haystack中找到了needle的第一个元素，然后判断之后的元素是否属于needle
                {
                    int count = 0;//记录循环比对次数
                    int k = 0; //指向needle的头元素
                    while (count<=needle_size)
                    {
                        //进入循环，比对该元素之后是否是needle所含元素,循环次数必不大于needle长度
                        if (haystack[i] != needle[k])//该元素之后的元素不属于needle
                        {
                            i = i - count + 1; //将i返回，指向haystack中下一个元素
                            break;//跳出此循环，去haystack中寻找下一个与needle头元素相同的元素下标
                        }
                        i += 1;//指针各加一，判断下一对儿元素是否一致
                        k += 1;
                        count += 1;//循环次数+1
                        if (count == needle_size)//循环次数=needle长度，则说明一一对应，此时i指向尾巴元素
                            return i - needle_size;
                        
                    }
                }
            }
            return -1;
        }
    }

};

int main()
{
    string haystack = "mississippi";
    string needle = "issip";
    Solution sol;
    int p;
    p = sol.strStr(haystack, needle);
    cout << p << endl;
    int u;
    cin >> u;
    return 0;
}