567. 字符串的排列
给定两个字符串 s1 和 s2,写一个函数来判断 s2 是否包含 s1 的排列。
换句话说,第一个字符串的排列之一是第二个字符串的子串。
示例1:
输入: s1 = “ab” s2 = “eidbaooo”
输出: True
解释: s2 包含 s1 的排列之一 (“ba”).
示例2:
输入: s1= “ab” s2 = “eidboaoo”
输出: False
注意:
输入的字符串只包含小写字母
两个字符串的长度都在 [1, 10,000] 之间
解题思路:
滑动窗口 + 字典
定义一个长度与S1相同的滑动窗口,可以是数组也可以是字典。本人用的是字典,滑动窗口放在S2上,left为S2的最左端点开始向右步长为1开始滑动,定义 counter1 是对 s1 内字符出现的个数的统计,定义 counter2 是对 s2 内字符出现的个数的统计。在窗口每次右移的时候,需要把右边新加入窗口的字符个数在 counter2 中加 1,把左边移出窗口的字符的个数减 1。如果 counter1 == counter2 ,那么说明窗口内的子串是 s1 的一个排列,返回 True;如果窗口已经把 s2 遍历完了仍然没有找到满足条件的排列,返回 False。
class Solution:
def checkInclusion(self, s1: str, s2: str) -> bool:
# 统计 s1 中每个字符出现的次数
N=len(s2)
# 定义滑动窗口的范围是 [left, right],闭区间,长度与s1相等
left=0
right=len(s1)-1
# 统计窗口s1内的元素出现的次数
counter1 = collections.Counter(s1)
# 统计窗口s2[left, right - 1]内的元素出现的次数
counter2 = collections.Counter(s2[0:right])
while right<N:
counter2[s2[right]]+=1
if counter1==counter2:
return True
counter2[s2[left]]-=1
# 如果当前 left 位置的元素出现次数为 0, 需要从字典中删除,否则这个出现次数为 0 的元素会影响两 counter 之间的比较
if counter2[s2[left]]==0:
del counter2[s2[left]]
left+=1
right+=1
return False
本题中的 counter 可以用字典,也可以用数组来实现。用字典的时候,需要注意:如果移除 left 元素后,若 counter2[s2[left]] == 0 那么需要从字典中删除 left 这个key。因为 {“a”:0, “b”:1} 和 {“b”:1} 是不等的。
counter2 初始化的时候只放了 [0, right - 1] 个元素,因为在 while 循环中的第一行就是把 right 元素放到 counter2 中。