go语言LeeCode刷题记：30. 串联所有单词的子串

30. 串联所有单词的子串

给定一个字符串 s 和一些长度相同的单词 words。找出 s 中恰好可以由 words 中所有单词串联形成的子串的起始位置。

注意子串要与 words 中的单词完全匹配，中间不能有其他字符，但不需要考虑 words 中单词串联的顺序。

示例 1：

输入：
  s = "barfoothefoobarman",
  words = ["foo","bar"]
输出：[0,9]
解释：
从索引 0 和 9 开始的子串分别是 "barfoo" 和 "foobar" 。
输出的顺序不重要, [9,0] 也是有效答案。

示例 2：

输入：
  s = "wordgoodgoodgoodbestword",
  words = ["word","good","best","word"]
输出：[]

答案：

func findSubstring(s string, words []string) []int {
	if len(words) < 1 {
		return []int{}
	}

	slen,wlen,arrLen := len(s),len(words),len(words[0])
	if slen < arrLen*wlen {
		return []int{}
	}

	var res []int
	var mp = make(map[string]int)
	for _, w := range words {
		mp[w]++
	}

	for i := 0; i < arrLen; i++ {
		var count int
		var counterMap = make(map[string]int)

		for l, r := i, i; r <= slen-arrLen; r = r + arrLen {
			word := s[r : r+arrLen]
			if num, found := mp[word]; found {
				for counterMap[word] >= num {
					counterMap[s[l:l+arrLen]]--
					count--
					l += arrLen
				}
				counterMap[word]++
				count++
			} else {
				for l < r {
					counterMap[s[l:l+arrLen]]--
					count--
					l += arrLen
				}
				l += arrLen
			}
			if count == wlen {
				res = append(res, l)
			}
		}
	}

	return res
}