swift 實現strStr() - LeetCode
阿新 • • 發佈:2018-11-07
實現 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 { func strStr(_ haystack: String, _ needle: String) -> Int { if needle.isEmpty { return 0 } let hLength = haystack.count let nLength = needle.count if hLength < nLength { return -1 } let threshold = hLength - nLength for i in 0...threshold { let subStr = haystack[haystack.index(haystack.startIndex, offsetBy: i)..<haystack.index(haystack.startIndex, offsetBy: i + nLength)] if subStr == needle { return i } } return -1 } }