438. 找到字符串中所有字母异位词
给定两个字符串 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” 的异位词。
解题思路
滑动窗口 + 双指针
除了直接比较数组是否相等外,其实还可以用双指针来表示滑动窗口的两侧边界,当滑动窗口的长度等于p的长度时,表示找到一个异位词,两种方式的时间复杂度都是O(n)级别的
先说结论,Python用数组更快一点点(差不太多其实),Java用双指针更快一点,下面是具体步骤:
定义滑动窗口的左右两个指针left,right
right一步一步向右走遍历s字符串
right当前遍历到的字符加入s_cnt后不满足p_cnt的字符数量要求,将滑动窗口左侧字符不断弹出,也就是left不断右移,直到符合要求为止。
当滑动窗口的长度等于p的长度时,这时的s子字符串就是p的异位词。
其中,left和right表示滑动窗口在字符串s中的索引,cur_left和cur_right表示字符串s中索引为left和right的字符在数组中的索引
def findAnagrams1(s, p):
list_s = [0] * 26
list_p = [0] * 26
res = []
for i in range(len(p)):
list_p[ord(p[i]) - ord('a')] += 1
for i in range(len(s)):
list_s[ord(s[i]) - ord('a')] += 1
if i < len(p) - 1:
continue
if list_s == list_p:
res.append(i - len(p) + 1)
list_s[ord(s[i - len(p) + 1]) - ord('a')] -= 1
return res