找到字符串中所有字母异位词

GitHub跳转原题关系图返回列表

固定长度滑动窗口维护 26 维字母计数,窗口进出各一次,O(n)。

OJ: leetcodecn

题目 ID: find-all-anagrams-in-a-string

难度:普及+/提高

标签:哈希表字符串滑动窗口cpppython

日期: 2026-07-28 22:05

题意

给定字符串 s 和 p,找出 s 中所有 p 的字母异位词子串的起始下标。

思路

暴力排序每个窗口 O(nk log k)。优化:用 26 维计数器维护窗口内字符频次,每次窗口滑动时只更新进出两个字符,然后比较整个计数数组是否归零。

代码

cpp
/**
 * Author by Rainboy
 */
// main.cpp:滑动窗口维护 26 维计数,O(n)。
#include <bits/stdc++.h>
using namespace std;

class Solution {
public:
    vector<int> findAnagrams(string s, string p) {
        int n = s.size(), m = p.size();
        if (n < m)
            return {};
        array<int, 26> cnt = {};
        for (char ch : p)
            cnt[ch - 'a']++;
        vector<int> ans;
        for (int i = 0; i < n; i++) {
            cnt[s[i] - 'a']--;
            if (i >= m)
                cnt[s[i - m] - 'a']++;
            if (i >= m - 1) {
                bool ok = true;
                for (int x : cnt)
                    if (x) {
                        ok = false;
                        break;
                    }
                if (ok)
                    ans.push_back(i - m + 1);
            }
        }
        return ans;
    }
};

int main() {
    ios::sync_with_stdio(false);
    cin.tie(nullptr);
    string s, p;
    cin >> s >> p;
    auto v = Solution().findAnagrams(s, p);
    for (int x : v)
        cout << x << ' ';
    return 0;
}
python
#!/usr/bin/env python3
from typing import List


class Solution:
    def findAnagrams(self, s: str, p: str) -> List[int]:
        n, m = len(s), len(p)
        if n < m:
            return []
        cnt = [0] * 26
        for ch in p:
            cnt[ord(ch) - 97] += 1
        ans = []
        for i in range(n):
            cnt[ord(s[i]) - 97] -= 1
            if i >= m:
                cnt[ord(s[i - m]) - 97] += 1
            if i >= m - 1:
                if all(x == 0 for x in cnt):
                    ans.append(i - m + 1)
        return ans


def main() -> None:
    s = input().strip()
    p = input().strip()
    ans = Solution().findAnagrams(s, p)
    print(*ans)


if __name__ == "__main__":
    main()

复杂度

  • 时间复杂度:O(n),每个字符进出各一次,每次检查 26 个计数(常数)。
  • 空间复杂度:O(1),固定 26 长度的计数数组。

总结

固定长度滑动窗口配合计数数组做"异位词匹配",是字符串模式匹配的经典模型。窗口滑动时只维护变化的两端,将每次匹配的时间从 O(k log k) 降到 O(1)。