给定两个字符串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” 的异位词。
核心思路:
滑动窗口的基本思路是维护一个长度为p的窗口,在字符串s中滑动这个窗口,并检查窗口中的子串是否为字符串p的异位词。异位词本质就是出现字母的频率是相同的。
#include <iostream>
#include <vector>
#include <string>
using namespace std;
class Solution
{
public:
vector<int> findAnagrams(string s, string p)
{
vector<int> result;
if (s.size() < p.size())
return result;
//p_count记录p中各字符的频率,s_count用来记录每个窗口变化的字符频率
vector<int> p_count(26, 0), s_count(26, 0);
for (int i = 0; i < p.size(); i++)
{
p_count[p[i] - 'a']++;
s_count[s[i] - 'a']++;
}
// 滑动窗口
for (int i = p.size(); i < s.size(); i++)
{
if (p_count == s_count)
result.push_back(i - p.size());
// 更新窗口
s_count[s[i] - 'a']++;
// 移除旧的元素
s_count[s[i - p.size()] - 'a']--;
}
// 检查最后最后一个窗口
if (p_count == s_count)
result.push_back(s.size() - p.size());
return result;
}
};
int main()
{
Solution solution;
string s = "cbaebabacd";
string p = "abc";
vector<int> result = solution.findAnagrams(s, p);
cout << "Anagram indices: ";
for (int index : result)
{
cout << index << " ";
}
cout << endl;
return 0;
}