题目描述:
给定两个字符串
s
和p
,找出s
中所有p
的字母异位词的起始索引。返回的答案以数组形式表示。
字母异位词定义:
若两个字符串包含的字符种类和出现次数完全相同,顺序无所谓,则互为字母异位词。
例如:"abc"
和 "bca"
是异位词;"aab"
和 "aba"
也是。
基本思路:
p
和 s
某个长度为 p.length()
的子串的字母频次。p
的一个异位词。class Solution {
public:
vector<int> findAnagrams(string s, string p) {
vector<int> ans;
if (s.length() < p.length()) return ans;
array<int, 26> cnt_p{}; // 存 p 中每个字母的出现次数
array<int, 26> cnt_s{}; // 存 s 当前窗口中每个字母的出现次数
for (char c : p) {
cnt_p[c - 'a']++;
}
for (int i = 0; i < s.length(); ++i) {
cnt_s[s[i] - 'a']++; // 当前字符进入窗口
if (i >= p.length()) {
// 超出窗口长度,移除最左字符
cnt_s[s[i - p.length()] - 'a']--;
}
if (cnt_s == cnt_p) {
// 当前窗口字符频次与 p 相同,记录左边界
ans.push_back(i - p.length() + 1);
}
}
return ans;
}
};
核心思想:
cnt[26]
,初始记录 p
中每个字符出现次数。cnt
中所有值为 0。class Solution {
public:
vector<int> findAnagrams(string s, string p) {
vector<int> ans;
if (s.length() < p.length()) return ans;
int cnt[26] = {0}; // 差值数组
// 初始化 p 的频次
for (char c : p) {
cnt[c - 'a']++;
}
int left = 0;
for (int right = 0; right < s.length(); ++right) {
int idx = s[right] - 'a';
cnt[idx]--; // 当前字符进入窗口
// 如果某个字母用得太多,收缩左边界
while (cnt[idx] < 0) {
cnt[s[left] - 'a']++;
left++;
}
// 若窗口长度等于 p 的长度,则说明是异位词
if (right - left + 1 == p.length()) {
ans.push_back(left);
}
}
return ans;
}
};
每次移动窗口,都:
p.length()
,若是且匹配,则记录起始索引。数组比 unordered_map 更快
,因为操作字符时用 char - 'a'
转索引非常高效。解法 | 原理 | 时间复杂度 | 空间复杂度 | 特点 |
---|---|---|---|---|
解法一 | 滑动窗口 + 频次比较 | O(n) | O(1) | 简洁直观 |
解法二 | 差值滑动窗口 | O(n) | O(1) | 更快的收缩 |