438. 找到字符串中所有字母异位词-Medium.md

Description

给定一个字符串 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" 的字母异位词。

Solution

• 时间复杂度：$O(n)$
• 空间复杂度：$O(n)$

class Solution:
    def findAnagrams(self, s: str, p: str) -> List[int]:
        need, window = defaultdict(int), defaultdict(int)
        left, right = 0, 0
        valid = 0
        ans = []

        for i in p:
            need[i] += 1

        while right < len(s):
            tmp_right = s[right]
            right += 1

            # window moves right.
            if tmp_right in need:
                window[tmp_right] += 1
                if window[tmp_right] == need[tmp_right]:
                    valid += 1

            # window moves left.
            while valid == len(need):
                if right - left == len(p):
                    ans.append(left)

                tmp_left = s[left]
                left += 1

                if tmp_left in need:
                    if window[tmp_left] == need[tmp_left]:
                        valid -= 1
                    window[tmp_left] -= 1

        return ans