题目:
给定两个字符串 s 和 p,找到 s 中所有 p 的 异位词 的子串,返回这些子串的起始索引。不考虑答案输出的顺序。
异位词 指字母相同,但排列不同的字符串。
示例:
输入: s = “cbaebabacd”, p = “abc”
输出: [0,6]
解释: 起始索引等于 0 的子串是 “cba”, 它是 “abc” 的异位词。 起始索引等于 6 的子串是 “bac”, 它是 “abc” 的异位词。
思路:
滑动窗口,
因为要找的是异位词,所以出现顺序可以不一样,所以可以统计每个字母出现的次数,先定义两个数组,分别表示p的字母次数情况,各个窗口的字母次数情况,初始化为0,先遍历一个窗口大小的子串,对p和s的窗口大小子串计数,才能进行后面的操作。这里如果两个数组就相等了,在结果数组中加入第一个索引0。
ord()函数返回字符的ASC码,减去字符a的ASC码可以使索引从0开始,对应上计数数组。
后边将窗口左边的字母数减1,右边新遍历到的字母数加1,固定窗口的长度,判断当前窗口的计数和pc是否相等,如果相等将起始索引加进结果里。遍历完字符串,返回结果。
时间复杂度是O(n),空间复杂度是O(1)。
class Solution(object):
def findAnagrams(self, s, p):
"""
:type s: str
:type p: str
:rtype: List[int]
"""
m = len(s)
n = len(p)
res = []
if m < n:return res #注意特殊情况
pc = [0] * 26
sc = [0] * 26 #这里要是分开的两个变量
for i in range(n):
pc[ord(p[i])-ord('a')] += 1
sc[ord(s[i])-ord('a')] += 1
if pc == sc:
res.append(0)
for i in range(n,m):
sc[ord(s[i-n])-ord('a')] -= 1
sc[ord(s[i])-ord('a')] += 1
if sc == pc:
res.append(i - n + 1)
return res