题目描述
给定一个字符串 s 和一个非空字符串 p,找到 s 中所有是 p 的字母异位词的子串,返回这些子串的起始索引。
字符串只包含小写英文字母,并且字符串 s 和 p 的长度都不超过 20100。
说明:
- 字母异位词指字母相同,但排列不同的字符串。
- 不考虑答案输出的顺序。
示例 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" 的字母异位词。
问题分析
此题用滑动窗口解决。先判断如果字符串s的长度小于字符串p的长度,那么返回{}。然后创建保存答案的数组ans,创建保存字符串s中字符出现次数的数组preq_s,大小为26,初始值为0。创建保存字符串p中字符出现次数的数组preq_p,大小为26,初始值为0。然后for循环p.size()次,把字符串s的前p.size()个字符的出现次数保存在数组preq_s中,把字符串p的前p.size()个字符的出现次数保存在数组preq_p中。然后判断如果数组preq_s和数组preq_p相同,则将0加入ans中。接下来开始滑动窗口,先定义窗口的左边界为 l ,初始值为0,定义窗口的右边界的下一个位置为 r ,初始值为p.size()。如果 r 在字符串s内,那么将窗口向右移动一个单位:将 r 位置的字符s[r]的出现次数在数组preq_s中加1,然后将 l 位置的字符s[l]的出现次数在数组preq_s中减1。然后将 r 和 l 都加1。然后判断如果数组preq_s和数组preq_p相同,则将 l 加入ans中。如此进行,当滑动窗口的右边界的下一个位置 r不在字符串s中时,这时滑动窗口已经顶到了字符串s的右边界,所以退出while循环,返回ans。
代码实现
class Solution {
public:
vector<int> findAnagrams(string s, string p) {
if(s.size() < p.size())
return {};
vector<int> ans;
vector<int> preq_s(26, 0), preq_p(26, 0);
for(int i = 0; i < p.size(); i++){
++preq_s[s[i] - 'a'];
++preq_p[p[i] - 'a'];
}
if(preq_s == preq_p)
ans.push_back(0);
int l = 0;
int r = p.size();
while(r < s.size()){
++preq_s[s[r++] - 'a'];
--preq_s[s[l++] - 'a'];
if(preq_s == preq_p)
ans.push_back(l);
}
return ans;
}
};