题目
You are given a string, s, and a list of words, words, that are all of the same length. Find all starting indices of substring(s) in s that is a concatenation of each word in words exactly once and without any intervening characters.
Example 1:
Input:
s = "barfoothefoobarman",
words = ["foo","bar"]
Output: [0,9]
Explanation: Substrings starting at index 0 and 9 are "barfoor" and "foobar" respectively.
The output order does not matter, returning [9,0] is fine too.
Example 2:
Input:
s = "wordgoodgoodgoodbestword",
words = ["word","good","best","word"]
Output: []
我的代码
package main
import "fmt"
func findSubstring(s string, words []string) []int {
if words == nil || len(words) == 0 {
return nil
}
var result = make([]int, 0)
wordsMap := sliceToMap(words)
wordLen := len(words[0])
slen := len(s)
temMap := make(map[string]int)
for i := 0; i <= slen-wordLen*len(words); i++ {
copyMap(wordsMap, temMap)
matchWord := len(words)
for j := i; j <= slen-wordLen; j += wordLen {
word := s[j : j+wordLen]
if value, ok := temMap[word]; ok && value > 0 {
temMap[word] = value - 1
matchWord--
if matchWord == 0 {
result = append(result, i)
break
}
} else {
break
}
}
}
return result
}
func sliceToMap(words []string) map[string]int {
var wordsMap = make(map[string]int)
for _, word := range words {
wordsMap[word] = wordsMap[word] + 1
}
return wordsMap
}
func copyMap(src, target map[string]int) {
for key, value := range src {
target[key] = value
}
}
func main() {
fmt.Println(findSubstring("barfoothefoobarman", []string{"foo", "bar"}))
fmt.Println(findSubstring("wordgoodgoodgoodbestword", []string{"word", "good", "best", "good"}))
fmt.Println(findSubstring("foobarfoobar", []string{"foo", "bar"}))
}