LeetCode 438. 找到字符串中所有字母异位词

438. 找到字符串中所有字母异位词

给定两个字符串 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 仅包含小写字母

 

思路：

• 先构建两个数组 sCnt 和 pCnt，分别统计【在s中滑动的滑动窗口】和【被比较字符串p】中字母出现的次数
• 正式对两个数组比较之前，首先要根据字符串p来填充一次 sCnt 和 pCnt，并判断是否为字母异位词
• 后续只需更新滑动窗口并比较，不断使其右移即可（左出右进，保证滑动窗口长度为 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 和滑动窗口中每种字母的数量。

func findAnagrams(s string, p string) []int {
    res := make([]int, 0)
    pLen, sLen := len(p), len(s)
    
    if sLen < pLen {
        return res
    }

    // 分别统计【在s中滑动的滑动窗口】和【被比较字符串p】中字母出现的次数
    sCnt, pCnt := [26]int{}, [26]int{} 

    // 当滑动窗口的首位在s[0]处时 （相当于首次放置滑动窗口进入数组）
    for i := 0; i < len(p); i++ {
        pCnt[p[i]-'a'] += 1
        sCnt[s[i]-'a'] += 1
    }

    // 首次比较 s[:pLen] 与 p 是否为字母异位词
    if sCnt == pCnt {
        res = append(res, 0)
    }

    for i := 0; i < len(s) - pLen; i++ {
        // 更新滑动窗口，不断右移
        sCnt[s[i]-'a']--        // (1) 先删除当前滑动窗口中首位的字母记录
        sCnt[s[i+pLen]-'a']++   // (2) 再从当前滑动窗口右侧加入一个最新的字母记录

        // 每次窗口滑动后，判断是否满足字母异位词条件
        if sCnt == pCnt {
            res = append(res, i+1) // 注意:这里下标为i的字母之前已删除，所以实质上是从i+1开始比较
        }
    }

    return res
}