1、题目描述
https://leetcode-cn.com/problems/permutation-in-string/
给定两个字符串 s1 和 s2,写一个函数来判断 s2 是否包含 s1 的排列。
换句话说,第一个字符串的排列之一是第二个字符串的子串。
- 输入的字符串只包含小写字母
- 两个字符串的长度都在 [1, 10,000] 之间
输入: s1 = "ab" s2 = "eidbaooo"
输出: True
解释: s2 包含 s1 的排列之一 ("ba").
输入: s1= "ab" s2 = "eidboaoo"
输出: False
2、代码详解
滑动窗口 + 字典
- 使用一个长度和 s1 长度相等的固定窗口大小的滑动窗口,在 s2 上面从左向右滑动
- 判断 s2 在滑动窗口内的每个字符出现的个数是否跟 s1 每个字符出现次数完全相等
import collections
class Solution:
def checkInclusion(self, s1: str, s2: str) -> bool:
counter1 = collections.Counter(s1) # s1 中每个字符出现的次数
n = len(s2)
# 滑动窗口的范围是 [left, right],闭区间,长度与s1相等
left = 0
right = len(s1) - 1
# 统计窗口s2[left, right - 1]内的元素出现的次数
counter2 = collections.Counter(s2[0:right])
while right < n:
counter2[s2[right]] += 1
if counter1 == counter2:
return True
# 窗口向右移动前,把当前 left 位置的元素出现次数 - 1
counter2[s2[left]] -= 1
# 当前 left 位置的元素次数 0, 需从字典中删除,否则这个出现次数为 0 的元素会影响两 counter 之间的比较
# 因为{"a":0, "b":1} 和 {"b":1} 是不等的
if counter2[s2[left]] == 0:
del counter2[s2[left]]
left += 1
right += 1
return False
s1 = "ab"
s2 = "eidbaooo"
s = Solution()
print(s.checkInclusion(s1, s2))