leetcode 28. 實現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() 定義相符。

解法一：

class Solution {
public:
    int strStr(string haystack, string needle) {
        if(needle == "") return 0;
        if(haystack.size() < needle.size()) return -1;
        int p;
        for(int i = 0;i < haystack.size()-needle.size()+1;++i){
            if(haystack.substr(i,needle.size()) == needle)
                return i;
        }
        return -1;
    }
};

解法二：

class Solution {
public:
    int strStr(string haystack, string needle) {
        if(needle == "") return 0;
        if(haystack.size() < needle.size()) return -1;
        int flag = 1;
        int p;
        for(int i = 0;i < haystack.size()-needle.size()+1;++i){
            p = i;
            for(int j = 0;j < needle.size();++j){
                if(haystack[p++] != needle[j]){
                    flag = 0;
                    break;
                }
                flag = 1;
            }
            if(flag == 1) return i;
        }
        return -1;
    }
};

認真審題！