字符串排列
给定两个字符串 s1 和 s2,写一个函数来判断 s2 是否包含 s1 的排列。
换句话说,第一个字符串的排列之一是第二个字符串的 子串 。
思路: 维护26格char计数数组,全部为0时返回True,diff变量记录不同char个数,滑动窗口遍历数组,首先从起点出发,计算当前charcount和diff,然后每次只考虑左右位置,对左右位置对应char的count进行操作,最后若diff为0返回True
class Solution:
def checkInclusion(self, s1: str, s2: str) -> bool:
if len(s2) < len(s1):
return False
char_list = [0] * 26
for i in range(len(s1)):
char_list[ord(s1[i])-ord('a')] += 1
char_list[ord(s2[i])-ord('a')] -= 1
diff = 0
for c in char_list:
if c != 0:
diff += 1
if diff == 0:
return True
for i in range(len(s1), len(s2)):
left, right = ord(s2[i - len(s1)]) - ord('a'), ord(s2[i]) - ord('a')
if left == right:
continue
if char_list[right] == 0:
diff += 1
char_list[right] -= 1
if char_list[right] == 0:
diff -= 1
if char_list[left] == 0:
diff += 1
char_list[left] += 1
if char_list[left] == 0:
diff -= 1
if diff == 0:
return True
return False
思路: 维护一个26格的计数数组,遍历s1获得计数,然后双指针遍历s2,减去遍历到的字母,如果遇到过多char,右移左指针,最后如果两个指针长度差为s1长度,返回True
class Solution:
def checkInclusion(self, s1: str, s2: str) -> bool:
char_list = [0] * 26
for c in s1:
char_list[ord(c)-ord('a')] += 1
left = 0
for right in range(len(s2)):
char_list[ord(s2[right])-ord('a')] -= 1
while char_list[ord(s2[right])-ord('a')] < 0:
char_list[ord(s2[left])-ord('a')] += 1
left += 1
if right - left + 1 == len(s1):
return True
return False