给定一个字符串 s 和一个非空字符串 p,找到 s 中所有是 p 的字母异位词的子串,返回这些子串的起始索引。
字符串只包含小写英文字母,并且字符串 s 和 p 的长度都不超过 20100。
说明:
字母异位词指字母相同,但排列不同的字符串。
不考虑答案输出的顺序。
示例 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" 的字母异位词。
链接:https://leetcode-cn.com/problems/find-all-anagrams-in-a-string
这道题稍有难度。用暴力循环遍历就会超时:
class Solution:
def findAnagrams(self, s: str, p: str) -> List[int]:
res = []
lenp,lens = len(p),len(s)
ls2 = sorted(p)
for i in range(lens-lenp+1):
if sorted(s[i:i+lenp]) == ls2:
res.append(i)
return res
借鉴了一个大神的代码,用双指针+滑动窗口:
class Solution:
def findAnagrams(self, s: str, p: str) -> List[int]:
res = []
window = {} # 记录窗口中各个字符数量的字典
needs = {} # 记录目标字符串中各个字符数量的字典
for ch in p:
needs[ch] = needs.get(ch, 0) + 1 # 统计目标字符串的信息
lenp, lens = len(p), len(s)
left = right = 0 # 两个指针,分别表示窗口的左、右界限
while right < lens:
ch = s[right]
if ch not in needs: # 当遇到不需要的字符时
window.clear() # 将之前统计的信息全部放弃
left = right = right + 1 # 从下一位置开始重新统计
else:
window[ch] = window.get(ch, 0) + 1 # 统计窗口内各种字符出现的次数
if right-left+1 == lenp: # 当窗口大小与目标字符串长度一致时
if window == needs:
res.append(left) # 如果窗口内的各字符数量与目标字符串一致就将left添加到结果中
window[s[left]] -= 1 # 并将移除的字符数量减一
left += 1
right += 1 # right右移
return res
Python实现大佬滑动窗口思想,速度击败99.32% - 找到字符串中所有字母异位词 - 力扣(LeetCode) https://leetcode-cn.com/problems/find-all-anagrams-in-a-string/solution/pythonshi-xian-da-lao-hua-dong-chuang-kou-si-xiang/