给定两个字符串 s1 和 s2,写一个函数来判断 s2 是否包含 s1 的排列。
换句话说,第一个字符串的排列之一是第二个字符串的子串。
示例1:
输入: s1 = "ab" s2 = "eidbaooo"
输出: True
解释: s2 包含 s1 的排列之一 ("ba").
示例2:
输入: s1= "ab" s2 = "eidboaoo"
输出: False
注意:
- 输入的字符串只包含小写字母
- 两个字符串的长度都在 [1, 10,000] 之间
解题思路:
这道题涉及两个核心思路,1是字符串本身转成可比较的分布,2是滑动窗口。对于比较一个字符串是否在另外一个字符串排列中1的方法可以通用。对于一个集合每次只有部分改变的情况都可以考虑滑动窗口。
class Solution(object):
def checkInclusion(self, s1, s2):
"""
:type s1: str
:type s2: str
:rtype: bool
"""
def get_distribution(s):
start = ord('a')
container = [0]*26
for i in s:
#print(ord(i),i)
container[ord(i) - start ]+=1
return container
start = ord('a')
n = len(s1)
m = len(s2)
if n > m:
return False
d_s1 = get_distribution(s1)
d_s2 = get_distribution(s2[0:n])
if d_s1 == d_s2:
return True
for i in range(n,m):
#print(d_s1,d_s2)
d_s2[ ord(s2[i-n]) - start ] -= 1
d_s2[ ord(s2[i]) - start ] += 1
if d_s1 == d_s2:
return True
return False