给定两个字符串 s 和 p,找到 s 中所有 p 的 异位词 的子串,返回这些子串的起始索引。不考虑答案输出的顺序。
异位词 指由相同字母重排列形成的字符串(包括相同的字符串)。
本题我原本的思路是遍历s的下标,然后截取p长度的子串和p进行比较,但超出时间限制。
采用滑动窗口法的解法是,使用两个数组sCount来记录p长度的子串字符出现的次数,pCount记录p中的字符出现次数,遍历s字符串,每次将窗口向右移动一位,更改子串中字符出现次数,若两者相等则将下标加入结果数组中,代码如下:
class Solution {
public:
vector<int> findAnagrams(string s, string p) {
int len = p.size();
int size = s.size();
vector<int> res;
if(len > size) return res;
vector<int> sCount(26);
vector<int> pCount(26);
for(int i = 0; i < len; i++){
sCount[s[i] - 'a']++;
pCount[p[i] - 'a']++;
}
if(sCount == pCount) res.push_back(0);
for(int i = 0; i < size - len; i++) {
sCount[s[i] - 'a']--;
sCount[s[i + len] - 'a']++;
if(sCount == pCount){
res.push_back(i + 1);
}
}
return res;
}
};