思路:滑动窗口 + 数组
1.因为字符串中的字符全是小写字母,可以用长度为26的数组记录字母出现的次数。
2.plen =p.size(),记录p字符串的字母频次pcount,和s字符串前plen个字母频次(滑动窗口)scount。 若pcount和scount相等,则找到第一个异位词索引 0。
3.继续遍历s字符串,在scount中每次增加一个新字母s[i+plen],去除一个旧字母s[i]。
4.判断pcount和scount是否相等,相等则在返回值res中添加索引位置 i + 1。
class Solution {
public:
vector<int> findAnagrams(string s, string p) {
int plen =p.size(),slen=s.size();
if(slen<plen) {return vector<int>();}//s的长度小于字符串 p的长度时,返回空vector
//一定要考虑到特殊情况,不加的话,scount下标会出现负值,报错。
vector<int> res; //存答案
vector<int> pcount(26);
vector<int> scount(26);//利用滑动窗口,来对比s子串与p是否相同
for(int i=0;i<plen;i++){//记录p各字母数量,同时与s从0开始第一个滑动窗口对比(妙,否则要另遍历s滑动窗口)
pcount[p[i]-'a']++; //注意是p[i],不是s[i],也不是i
scount[s[i]-'a']++;
}
if(pcount==scount) res.push_back(0);
for(int i=0;i<slen-plen;i++){//s的滑动窗口依次后移,对比
scount[s[i]-'a']--;
scount[s[i+plen] -'a']++;
if(pcount==scount) res.push_back(i+1);//注意索引位置为i+1
}
return res;
}
};
注意:
1.滑动窗口真的很妙!!!
2.通过数组判断s 中所有p的异位词的子串!!但是要注意,使用数组来做哈希的题目,是因为题目都限制了数值的大小。
3.要考虑slen<plen特殊情况
复杂度分析
时间复杂度:O(n),for循环有O(n),数组的长度是常数,所以数组的比较也是常数级别的,那最终的时间复杂度就是O(n)。
空间复杂度:O(1),需要常数级别的额外空间。