Code11 实现 strStr()

题目

leetcode28. 实现 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() 定义相符。

代码

// C
#include <string.h>
int strStr(char* haystack, char* needle) {
  if (needle == "") {
    return 0;
  }

  int haystack_len = strlen(haystack);
  int needle_len= strlen(needle);
  char* temp = needle;
  for (int i = 0; i <= haystack_len - needle_len; ++i){
    int j = i;
    int k = 0;
    while((k < needle_len) && (haystack[j] == needle[k])){
      ++j;
      ++k;
    }

    if (k == needle_len){
      return i;
    }
  }

  return -1;
}

// C++
#include <string>
using namespace std;
class Solution {
 public:
  int strStr(string haystack, string needle) {
    if (needle.empty()) {
      return 0;
    }
    auto it = haystack.find(needle);
    if (it == string::npos) {
      return -1;
    } else {
      return it;
    }
  }
};

测试

#include <iostream>

int main() {
  {
    // C
    char* haystack = "hello";
    char* needle = "lo";
    cout << strStr(haystack, needle) << endl;
  }
  {
    // C++ 
    string haystack = "hello";
    string needle = "ll";
    Solution s;
    cout << s.strStr(haystack, needle);
  }
  cin.get();
  return 0;
}

• 结果

3
2