字符串的排列（LeetCode）

题目

        给你两个字符串 s1 和 s2 ，写一个函数来判断 s2 是否包含 s1 的排列。如果是，返回 true ；否则，返回 false 。

        换句话说，s1 的排列之一是 s2 的 子串 。

解题

from collections import Counter


def check_inclusion(s1: str, s2: str) -> bool:
    # s1长度大于s2时，直接返回False
    if len(s1) > len(s2):
        return False

    # 统计s1中每个字符的频率
    s1_count = Counter(s1)
    # 初始化窗口，大小为s1的长度
    window = Counter(s2[:len(s1)])

    # 如果初始窗口和s1的字符频率相同，返回True
    if window == s1_count:
        return True

    # 使用滑动窗口检查s2中是否存在s1的排列
    for i in range(len(s1), len(s2)):
        # 窗口右移，加入一个新的字符
        window[s2[i]] += 1
        # 移出窗口左侧的字符
        window[s2[i - len(s1)]] -= 1
        # 如果某个字符的计数变为0，从字典中删除
        if window[s2[i - len(s1)]] == 0:
            del window[s2[i - len(s1)]]

        # 如果窗口与s1的字符频率相同，返回True
        if window == s1_count:
            return True

    # 如果未找到符合条件的子串，返回False
    return False


# 示例
s1 = "ab"
s2 = "eidbaooo"
result = check_inclusion(s1, s2)
print(result)  # 输出: True