1. 程式人生 > 實用技巧 >LeetCode 567.字串的排列

LeetCode 567.字串的排列

給定兩個字串 s1 和 s2,寫一個函式來判斷 s2 是否包含 s1 的排列。
換句話說,第一個字串的排列之一是第二個字串的子串。

示例1:
輸入: s1 = "ab" s2 = "eidbaooo"
輸出: True
解釋: s2 包含 s1 的排列之一 ("ba").

示例2:
輸入: s1= "ab" s2 = "eidboaoo"
輸出: False

注意:
輸入的字串只包含小寫字母
兩個字串的長度都在 [1, 10,000] 之間

class Solution:
    def checkInclusion(self, s1: str, s2: str) -> bool:
        """
        分析:如果兩個字串的元素組成相同,則排列存在相同;
             問題轉化成:找s2中是否有子串與s1的元素組成相同,對於子串問題,這裡可以使用雙指標
             另外要注意把指標移動位置分析清楚
        """
        import numpy as np
        s1_lookup = np.array([0] * 26)
        s2_lookup = np.array([0] * 26)
        n = len(s1)
        for i in s1: 
            index = ord(i) - ord('a')
            s1_lookup[index] +=1
        #print(s1_lookup)
        L = 0
        for R in range(len(s2)):
            index = ord(s2[R]) - ord('a')          
            s2_lookup[index] += 1
            if R >= n:
                s2_lookup[ord(s2[L]) - ord('a')] -= 1
                L += 1
            #s2_substr = s2[L:R+1]
            #print(L,R,s2_substr,s2_lookup)
            for pos in range(len(s1_lookup)):                
                if s1_lookup[pos] != s2_lookup[pos]:
                    break            
            if pos+1 == len(s1_lookup):
                return True
        return False