forked from soapyigu/LeetCode-Swift
-
Notifications
You must be signed in to change notification settings - Fork 0
/
StrStr.swift
34 lines (30 loc) · 908 Bytes
/
StrStr.swift
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
/**
* Question Link: https://leetcode.com/problems/implement-strstr/
* Primary idea: Iterate two strings
* Time Complexity: O(nm), Space Complexity: O(n)
*/
class StrStr {
func strStr(_ haystack: String, _ needle: String) -> Int {
let hChars = Array(haystack.characters), nChars = Array(needle.characters)
let hLen = hChars.count, nLen = nChars.count
guard hLen >= nLen else {
return -1
}
guard nLen != 0 else {
return 0
}
for i in 0 ... hLen - nLen {
if hChars[i] == nChars[0] {
for j in 0 ..< nLen {
if hChars[i + j] != nChars[j] {
break
}
if j + 1 == nLen {
return i
}
}
}
}
return -1
}
}