题目
给你两个字符串
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