题目
给定两个字符串 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” 的异位词。
答案
以下是使用 Python 实现找到字符串 s 中所有 p 的异位词的子串,并返回这些子串的起始索引的代码:
def findAnagrams(s, p):""":type s: str:type p: str:rtype: List[int]"""# 初始化两个字典,分别记录字符串 s 和 p 中每个字符出现的次数count_s = {}count_p = {}result = []# 遍历字符串 p,记录每个字符出现的次数for char in p:count_p[char] = count_p.get(char, 0) + 1# 遍历字符串 s,记录每个字符出现的次数,并判断是否是异位词for i, char in enumerate(s):count_s[char] = count_s.get(char, 0) + 1# 如果当前字符是 p 中的字符,且出现次数与 p 中相同,则起始索引为 iif char in count_p and count_s[char] == count_p[char]:result.append(i)# 如果当前字符不是 p 中的字符,且出现次数为 0,则将对应字符在 count_s 中删除if char not in count_p and count_s[char] == 0:del count_s[char]return result
注释说明:
count_s
和count_p
分别记录字符串s
和p
中每个字符出现的次数,初始值都为空字典。- 遍历字符串
p
,将每个字符加入到count_p
字典中,并将出现次数加一。 - 遍历字符串
s
,记录每个字符出现的次数,并判断是否是异位词。如果是异位词,则将起始索引加入到result
列表中。 - 如果当前字符不是
p
中的字符,且出现次数为 0,则将对应字符在count_s
中删除。因为已经判断过这个字符不是异位词,所以不再需要记录它的出现次数。