1. 程式人生 > >C++ Leetcode初級演算法之實現strStr()

C++ Leetcode初級演算法之實現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) {
        int hSize = haystack.length();
        int nSize = needle.length();
        if(hSize < nSize)   return -1;
        if(haystack == needle) return 0;
        for(int i=0;i<=hSize-nSize;i++)
            if(haystack.substr(i,nSize) == needle)
                return i;
        return -1;
    }
};