给定两个字符串
s
和p
,找到s
中所有p
的 异位词 的子串,返回这些子串的起始索引。不考虑答案输出的顺序。异位词 指由相同字母重排列形成的字符串(包括相同的字符串)。
示例 1:
输入: s = "cbaebabacd", p = "abc" 输出: [0,6] 解释: 起始索引等于 0 的子串是 "cba", 它是 "abc" 的异位词。 起始索引等于 6 的子串是 "bac", 它是 "abc" 的异位词。示例 2:
输入: s = "abab", p = "ab" 输出: [0,1,2] 解释: 起始索引等于 0 的子串是 "ab", 它是 "ab" 的异位词。 起始索引等于 1 的子串是 "ba", 它是 "ab" 的异位词。 起始索引等于 2 的子串是 "ab", 它是 "ab" 的异位词。提示:
1 <= s.length, p.length <= 3 * 104
s
和p
仅包含小写字母
思路:
将长度为len(p)且不断右移的滑动窗口sCntArr与pCntArr比较
- 先构建两个[26]int数组 sCntArr 和 pCntArr,分别统计"在s中滑动的滑动窗口"和"被比较字符串p"中字母出现的次数
- 正式比较两数组前,先要根据字符串p来填充1次sCntArr和pCntArr,并判断是否为异位词
- 后续只需更新滑动窗口并比较,不断使其右移即可(左出右进,保证滑动窗口长度为 len(p))
时间复杂度:O(m+(n−m)×Σ),其中 n 为字符串 s 的长度,m 为字符串 p 的长度,Σ 为所有可能的字符数。我们需要 O(m) 来统计字符串 p 中每种字母的数量;需要 O(m) 来初始化滑动窗口;需要判断 n−m+1 个滑动窗口中每种字母的数量是否与字符串 p 中每种字母的数量相同,每次判断需要O(Σ) 。因为 s 和 p 仅包含小写字母,所以 Σ=26。
空间复杂度: O(Σ),Σ=26。用于存储字符串 p 和滑动窗口中每种字母的数量。
/*
思路:
将长度为len(p)且不断右移的滑动窗口sCntArr与pCntArr比较
先构建两个[26]int数组 sCntArr 和 pCntArr,
分别统计"在s中滑动的滑动窗口"和"被比较字符串p"中字母出现的次数
正式比较两数组前,先要根据字符串p来填充1次sCntArr和pCntArr,并判断是否为异位词
后续只需更新滑动窗口并比较,不断使其右移即可(左出右进,保证滑动窗口长度为 len(p))
*/
func findAnagrams(s string, p string) []int {
res := make([]int, 0)
pLen, sLen := len(p), len(s)
if sLen < pLen {
return res
}
// 分别统计【在s中滑动的滑动窗口】和【被比较字符串p】中字母出现的次数
sCntArr, pCntArr := [26]int{}, [26]int{}
// 当滑动窗口的首位在s[0]处时 (相当于首次放置滑动窗口进入数组)
for i := 0; i < pLen; i++ {
pCntArr[p[i]-'a'] += 1
sCntArr[s[i]-'a'] += 1
}
// 首次比较 s[:pLen] 与 p 是否为字母异位词
if sCntArr == pCntArr {
res = append(res, 0)
}
for i := 0; i < sLen - pLen; i++ {
// 更新滑动窗口,不断右移
sCntArr[s[i]-'a']-- // (1) 先删除当前滑动窗口中首位的字母记录
sCntArr[s[i+pLen]-'a']++ // (2) 再从当前滑动窗口右侧加入一个最新的字母记录
// 每次窗口滑动后,判断是否满足字母异位词条件
if sCntArr == pCntArr {
res = append(res, i+1) // 注意:这里下标为i的字母之前已删除,所以实质上是从i+1开始比较
}
}
return res
}