剑指Offer|LCR 014. 字符串的排列

LCR 014. 字符串的排列

给定两个字符串 s1 和 s2，写一个函数来判断 s2 是否包含 s1 的某个变位词。

换句话说，第一个字符串的排列之一是第二个字符串的 子串 。

示例 1：

输入: s1 = "ab" s2 = "eidbaooo"
输出: True
解释: s2 包含 s1 的排列之一 ("ba").

示例 2：

输入: s1= "ab" s2 = "eidboaoo"
输出: False

提示：

• 1 <= s1.length, s2.length <= 104
• s1 和 s2 仅包含小写字母

法1：滑动窗口

分析：

建立一个hash表，键是26个字母，对应值是各自出现的次数，为方便键0就代表a，1代表b这样。

看例子：s1 = "ab" s2 = "eidbaooo"

表格中没写的都是填充0。接着遍历后面的s2

i=2,遍历s2中的d,

countss2.charCodeAt(i) - 'a'.charCodeAt(0)--=countsd-a--=counts3--

countss2.charCodeAt(i - s1.length) - 'a'.charCodeAt(0)++=countse-a++=counts4++

i=3,遍历s2中的b

countss2.charCodeAt(i) - 'a'.charCodeAt(0)--=countsb-a--=counts1--

countss2.charCodeAt(i - s1.length) - 'a'.charCodeAt(0)++=countsi-a++=counts8++

i=4,遍历s2中的a

countss2.charCodeAt(i) - 'a'.charCodeAt(0)--=countsa-a--=counts0--

countss2.charCodeAt(i - s1.length) - 'a'.charCodeAt(0)++=countsd-a++=counts3++

到这，counts全都为0，就返回true

 var checkInclusion = function(s1, s2) {
    if (s2.length < s1.length)  return false;

    let counts = new Array(26).fill(0);
    // 初始填充 counts 数组
    for (let i = 0; i < s1.length; ++i) {
        counts[s1.charCodeAt(i) - 'a'.charCodeAt(0)]++; // s1 计数 ++
        counts[s2.charCodeAt(i) - 'a'.charCodeAt(0)]--; // s2 计数 --
    }
    // 检查是否已匹配
    if (areAllZero(counts)) {
        return true;
    }

    // 滑动窗口
    // 滑动窗口的大小始终为 s1.length。
    for (let i = s1.length; i < s2.length; ++i) {
        counts[s2.charCodeAt(i) - 'a'.charCodeAt(0)]--;
        counts[s2.charCodeAt(i - s1.length) - 'a'.charCodeAt(0)]++;

        if (areAllZero(counts)) {
            return true;
        }
    }
};

/**
 * 辅助函数：检查 counts 数组是否全部为零
 * @param {number[]} counts
 * @return {boolean}
 */
function areAllZero(counts) {
    for (let count of counts) {
        if (count !== 0) {
            return false;
        }
    }
    return true;
}