LeetCode算法题——实现strStr()

题目如下：
实现 strStr() 函数。

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

示例 1:
输入: haystack = “hello”, needle = “ll”
输出: 2

示例 2:
输入: haystack = “aaaaa”, needle = “bba”
输出: -1

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

初步代码如下：

#include
class Solution {
public:
    int strStr(string haystack, string needle) 
    {
        if(needle.empty())
            return 0;
        if(haystack.empty() || haystack.length() < needle.length())
            return -1;
        for (int i = 0; i < haystack.length(); i++)
        {
            int j = 0,tmp = 0;
            if (haystack[i] == needle[0]) //寻找与字符串2第一位相同的位置
            {
                tmp = i;
                for (j; j < needle.length(); j++)
                {
                    if (haystack[i] == needle[j])
                    {
                        i++;
                        continue;
                    }
                    else
                    {
                        i = tmp;
                        break;
                    }
                }
                if (j == needle.length()) //证明字符串2完成对比，全部符合
                    return tmp; 
            }
        }
        return -1;
    }
};

优化后的代码：

#include
class Solution {
public:
    int strStr(string haystack, string needle) 
    {
        if(needle.empty())
            return 0;
        int m = haystack.length(),n = needle.length();
        if(m < n)
            return -1;
        int j=0;
        for(int i=0;i<=m-n;i++)
        {
            for(j=0;j<n;j++)
            {
                if(haystack[i+j]!=needle[j])
                    break;  
            }
            if(j==n)
                return i;
        }
        return -1;
    }
};

总结：
做对比尽量只扫描一次字符串，减少循环次数，先把字符串长度进行赋值，减少函数运算次数，换一个思路，从对比判断字符串相同变成判断字符串不同，能够有效提高代码运行的效率。