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

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

法1：滑动窗口

分析：

看例子s = "abab", p = "ab"

如果p的长度 大于 s的长度，说明s中不存在字母异位词，则返回空列表

定义pCount,将p中每个字母的出现次数存一下，pCount[0]表示字母a在p中出现次数；

先对比索引0处，是不是字母异位词。

遍历s前面和p一样长度的字符，将这部分s中字母频数记录到windowCount,此时，如果pCount和windowCount各个元素值相同，则下标0是字母异位词，就将0加到结果中。

接着再遍历s在p.length后的字符串，

索引每忘右移一下，就将当前元素的频数加到windowCount中，也需减去当前索引-p.length下标元素频数去掉，这样保持每次计算的频数长度与p的长度想用，每次都比较pCount和windowCount各个元素值相同，相同则将i - p.length + 1加入结果。

时间复杂度：$O(n)$

空间复杂度：$O(n)$

var findAnagrams = function(s, p) {
    if (s.length < p.length) return [];

    let result = [];
    let pCount = new Array(26).fill(0); // 用来记录 p 中每个字符的频率
    let windowCount = new Array(26).fill(0); // 用来记录当前窗口中每个字符的频率

    // 将 p 中每个字符的频率记录到 pCount 数组中
    for (let i = 0; i < p.length; i++) {
        pCount[p.charCodeAt(i) - 'a'.charCodeAt(0)]++;
    }

    // 初始化滑动窗口的第一个窗口
    for (let i = 0; i < p.length; i++) {
        windowCount[s.charCodeAt(i) - 'a'.charCodeAt(0)]++;
    }

    // 如果第一个窗口是变位词，加入结果
    if (arraysAreEqual(pCount, windowCount)) {
        result.push(0);
    }

    // 滑动窗口遍历字符串 s
    for (let i = p.length; i < s.length; i++) {
        // 新加入的字符
        windowCount[s.charCodeAt(i) - 'a'.charCodeAt(0)]++;
        // 移除左边的字符
        windowCount[s.charCodeAt(i - p.length) - 'a'.charCodeAt(0)]--;

        // 检查当前窗口是否为变位词
        if (arraysAreEqual(pCount, windowCount)) {
            result.push(i - p.length + 1);
        }
    }
    return result;
};

// 辅助函数：检查两个频率数组是否相等
function arraysAreEqual(arr1, arr2) {
    for (let i = 0; i < 26; i++) {
        if (arr1[i] !== arr2[i]) {
            return false;
        }
    }
    return true;
}