力扣刷题日记15---剑指 Offer II 015. 字符串中的所有变位词
给定两个字符串 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 仅包含小写字母
这道题与上一道题很像,区别在于,上一道题找到就可以结束,直接返回 true;而这道题要继续遍历,直到找出所有的变位词下标。因此可以直接在上一题的代码上直接修改,要修改的部分比较少。本文就不重复写思路了,有需要的查看上篇文章中的思路,两题基本一样。
思路1:滑动窗口优化
代码如下:
class Solution {
public:
vector<int> findAnagrams(string s, string p) {
int size1 = s.size(), size2 = p.size();
if(size1 < size2)
return {};
vector<int> v(26, 0);
vector<int> ret;
for(int i = 0; i < size2; ++i){
--v[p[i] - 'a'];
++v[s[i] - 'a'];
}
if(IsZero(v))
ret.push_back(0);
for(int i = size2; i < size1; ++i){
++v[s[i] - 'a'];
--v[s[i - size2] - 'a'];
if(IsZero(v))
ret.push_back(i - size2 + 1);
}
return ret;
}
bool IsZero(vector<int>& v){
for(int& n : v){
if(n)
return false;
}
return true;
}
};
时间复杂度:O((s.size() - p.size) * 26),后边的乘 26 是判断 v 中元素是否全部为 0 的复杂度;空间复杂度:O(26)。
思路2:双指针
代码如下:
class Solution {
public:
vector<int> findAnagrams(string s, string p) {
int size1 = s.size(), size2 = p.size();
if(size1 < size2)
return {};
vector<int> v(26, 0);
int left = 0, right = 0;
vector<int> ret;
for(int i = 0; i < size2; ++i){
--v[p[i] - 'a'];
}
while(right < size1){
++v[s[right] - 'a'];
while(v[s[right] - 'a'] > 0){
--v[s[left++] - 'a'];
}
if(++right - left == size2)
ret.push_back(left);
}
return ret;
}
};
时间复杂度:O(s.size() + p.size());空间复杂度:O(26)。