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==haystack)
            return 0;
        const int N = haystack.size()-needle.size()+1;
        for(int i=0; i<N; ++i){
            int j = i;
            int k = 0;
            while(j < haystack.size() && k<needle.size() && haystack[j]==needle[k]){
                ++j;
                ++k;
            }
            if(k==needle.size())
                return i;
        }
        return -1;
    }
};