0438：找到字符串中所有字母异位词（★）

力扣第 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" 的异位词。

提示:

• 1 <= s.length, p.length <= 3 * 104
• s 和 p 仅包含小写字母

分析

• 遍历每个窗口，判断是否符合即可
• 维护一个 Counter 判断子串是否符合
• 判断 Counter 是否符合有个省时的方法
  • 新加一个变量 valid 维护有用字符的个数
  • 移动 j 后，如果 ct[s[j]]=ct0[s[j]]，s[j] 就是有用的，valid 增 1
  • 移动 i 前，如果 ct[s[i]]=ct0[s[i]]，s[i] 就是有用的，valid 减 1

解答

 1
 2
 3
 4
 5
 6
 7
 8
 9
10
11
12
13
14
15
16
17
18

class Solution:
    def findAnagrams(self, s: str, p: str) -> List[int]:
        m = len(p)
        ct0,ct = Counter(p),defaultdict(int)
        valid = 0
        res = []
        for j,c in enumerate(s):
            ct[c] += 1
            if ct[c]==ct0[c]:
                valid += 1
            if j>=m:
                old = s[j-m]
                if ct[old]==ct0[old]:
                    valid -= 1
                ct[old] -= 1
            if valid==len(ct0):
                res.append(j-m+1)
        return res

72 ms