題目:
給定兩個字符串?s
?和?p
,找到?s
?中所有?p
?的?異位詞?的子串,返回這些子串的起始索引。不考慮答案輸出的順序。
示例?1:
輸入: s = "cbaebabacd", p = "abc" 輸出: [0,6] 解釋: 起始索引等于 0 的子串是 "cba", 它是 "abc" 的異位詞。 起始索引等于 6 的子串是 "bac", 它是 "abc" 的異位詞。
?示例 2:
輸入: s = "abab", p = "ab" 輸出: [0,1,2] 解釋: 起始索引等于 0 的子串是 "ab", 它是 "ab" 的異位詞。 起始索引等于 1 的子串是 "ba", 它是 "ab" 的異位詞。 起始索引等于 2 的子串是 "ab", 它是 "ab" 的異位詞。
提示:
1 <= s.length, p.length <= 3 * 104
s
?和?p
?僅包含小寫字母
解題思路:?
?將滑動窗口的大小設置為p的長度,如果s的長度或者s的剩余長度小于p直接返回,否則將窗口內的字母進行排序,如果排序得到的結果是相同的就認為是異位詞
代碼:
class Solution:def findAnagrams(self, s: str, p: str): len_s, len_p = len(s), len(p)if len_s<len_p:return []p_sorted = sorted(p)outputs = []for i in range(len(s)):if len_s-i<len_p:return outputss_sorted = sorted(s[i:i+len(p)])if s_sorted==p_sorted:outputs.append(i)return outputs
官方代碼:
# 官方代碼
class Solution:def findAnagrams(self, s: str, p: str):s_len, p_len = len(s), len(p)# 如果s的長度比p的長度小,則不存在異位詞if s_len < p_len:return []ans = []# 子串中只包含小寫字母,記錄字母的個數count = [0] * 26for i in range(p_len):count[ord(s[i]) - 97] += 1count[ord(p[i]) - 97] -= 1# 判斷一共多少個字母存在differ = [c != 0 for c in count].count(True)if differ == 0:ans.append(0)for i in range(s_len - p_len):# 遍歷s得到字母,如果在count中該字母的數量為1,則認為當前s和p中該字母抵消掉了,所有存在的字母種類數-1if count[ord(s[i]) - 97] == 1: # 窗口中字母 s[i] 的數量與字符串 p 中的數量從不同變得相同differ -= 1# 遍歷s得到字母,如果在count中該字母的數量為0,則認為p中沒有這個字母而s中有這個字母,s的字母種類數+1elif count[ord(s[i]) - 97] == 0: # 窗口中字母 s[i] 的數量與字符串 p 中的數量從相同變得不同differ += 1# 遍歷過s的字母之后將字母的個數-1count[ord(s[i]) - 97] -= 1# 利用滑動窗口來保證字母的種類相同,如果窗口的字母的個數及字母的種類和p的字母的個數及字母的種類相同則認為是一個異位詞if count[ord(s[i + p_len]) - 97] == -1: # 窗口中字母 s[i+p_len] 的數量與字符串 p 中的數量從不同變得相同differ -= 1elif count[ord(s[i + p_len]) - 97] == 0: # 窗口中字母 s[i+p_len] 的數量與字符串 p 中的數量從相同變得不同differ += 1count[ord(s[i + p_len]) - 97] += 1if differ == 0:ans.append(i + 1)return ans