一.剑指 Offer II 015. 字符串中的所有变位词
1.题干及思路
给定两个字符串 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
仅包含小写字母
注意:本题与主站 438 题相同: https://leetcode-cn.com/problems/find-all-anagrams-in-a-string/
要点:滑动窗口
思路:滑动窗口配合哈希表,这里哈希表用vector(26)替代因为vector重载过==符号,如果两个哈希表每个位置都相同说明是同位词
2.C++实现
class Solution {
public:
vector<int> findAnagrams(string s, string p) {
vector<int> answer;
int n = p.length(), m = s.length();
if (n > m) {
return {};
}
vector<int> cnt1(26), cnt2(26);
for (int i = 0; i < n; ++i) {
++cnt1[p[i] - 'a'];
++cnt2[s[i] - 'a'];
}
if (cnt1 == cnt2) {
answer.emplace_back(0);
}
for (int i = n; i < m; ++i) {
++cnt2[s[i] - 'a'];
--cnt2[s[i - n] - 'a'];
if (cnt1 == cnt2) {
answer.emplace_back(i-p.size()+1);
}
}
return answer;
}
};