1. 程式人生 > >Leetcode(28)實現strStr()

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() 定義相符。


解題思路
要知道string的str.substr(n,m)方法,這個方法返回的是str的第n個字元到第m個字元的字串,只需要用這個函式提取子字串與目標字串進行比較就可以了。
程式碼如下

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