【LeetCode567】字符串的排列

1.题目描述

给定两个字符串 s1 和 s2，写一个函数来判断 s2 是否包含 s1 的排列。

换句话说，第一个字符串的排列之一是第二个字符串的子串。

示例1:

输入: s1 = "ab" s2 = "eidbaooo"
输出: True
解释: s2 包含 s1 的排列之一 ("ba").
 

示例2:

输入: s1= "ab" s2 = "eidboaoo"
输出: False
 

注意：

输入的字符串只包含小写字母
两个字符串的长度都在 [1, 10,000] 之间

2.解法

# 解法一：滑窗法
class Solution:
    def match(self, l1, l2):
        for i in range(len(l1)):
            if l1[i] != l2[i]:
                return False
        return True
        
    def checkInclusion(self, s1: str, s2: str) -> bool:
        # 方法一：滑窗法
        if len(s1) > len(s2):
            return False
        s1_stat = [0] * 26
        s2_stat = [0] * 26
        for i, char in enumerate(s1):
            s1_stat[ord(char) - ord('a')] += 1
            s2_stat[ord(s2[i]) - ord('a')] += 1
        for j in range(0, len(s2) - len(s1)):
            if self.match(s1_stat, s2_stat):
                return True
            s2_stat[ord(s2[j]) - ord('a')] -= 1
            s2_stat[ord(s2[j + len(s1)]) - ord('a')] += 1
        return self.match(s1_stat, s2_stat)
# 方法二：改进滑窗法
class Solution:
    def match_count(self, l1, l2):
        count = 0
        for i in range(len(l1)):
            if l1[i] == l2[i]:
                count += 1
        return count
    
    def checkInclusion(self, s1: str, s2: str) -> bool:
        if len(s1) > len(s2):
            return False
        s1_stat = [0] * 26
        s2_stat = [0] * 26
        for i, char in enumerate(s1):
            s1_stat[ord(char) - ord('a')] += 1
            s2_stat[ord(s2[i]) - ord('a')] += 1
        count = self.match_count(s1_stat, s2_stat)
        for j in range(0, len(s2) - len(s1)):
            if count == 26:
                return True
            left = ord(s2[j]) - ord('a')
            right = ord(s2[j + len(s1)]) - ord('a')
            s2_stat[right] += 1
            if s2_stat[right] == s1_stat[right]:
                count += 1
            elif s2_stat[right] == s1_stat[right] + 1:
                count -= 1
            s2_stat[left] -= 1
            if s2_stat[left] == s1_stat[left]:
                count += 1
            elif s2_stat[left] == s1_stat[left] - 1:
                count -= 1
        return count == 26