题目如下:
实现 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<stdio.h>
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<stdio.h>
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;
}
};
总结:
做对比尽量只扫描一次字符串,减少循环次数,先把字符串长度进行赋值,减少函数运算次数,换一个思路,从对比判断字符串相同变成判断字符串不同,能够有效提高代码运行的效率。