[LeetCode] Implement strStr() 實現strStr()函式

Grandyang發表於2015-06-29
LeetCode函式

 

Implement strStr().

Returns the index of the first occurrence of needle in haystack, or -1 if needle is not part of haystack.

Update (2014-11-02):
The signature of the function had been updated to return the index instead of the pointer. If you still see your function signature returns a char * or String, please click the reload button to reset your code definition.

 

這道題讓我們在一個字串中找另一個字串第一次出現的位置，那我們首先要做一些判斷，如果子字串為空，則返回0，如果子字串長度大於母字串長度，則返回-1。然後我們開始遍歷母字串，我們並不需要遍歷整個母字串，而是遍歷到剩下的長度和子字串相等的位置即可，這樣可以提高運算效率。然後對於每一個字元，我們都遍歷一遍子字串，一個一個字元的對應比較，如果對應位置有不等的，則跳出迴圈，如果一直都沒有跳出迴圈，則說明子字串出現了，則返回起始位置即可，程式碼如下：

 

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

 

LeetCode All in One 題目講解彙總(持續更新中...)

相關文章