找到字符串中所有字母异位词
题目链接: 找到字符串中所有字母异位词
有关题目
给定两个字符串 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 * 10^4
s 和 p 仅包含小写字母
题解
法一:滑动窗口
参考官方题解
class Solution {
public:
vector<int> findAnagrams(string s, string p) {
int sLen = s.size(), pLen = p.size();
if (sLen < pLen)
return vector<int>();
vector<int> ans;
vector<int> sCnt(26);
vector<int> pCnt(26);
for (int i = 0; i < pLen; i++)
{
++sCnt[s[i] - 'a'];
++pCnt[p[i] - 'a'];
}
//sLen = pLen等长度的判断
if (sCnt == pCnt)
ans.emplace_back(0);
//滑动窗口
for (int i = 0; i < sLen - pLen; i++)
{
--sCnt[s[i] - 'a'];
++sCnt[s[i + pLen] - 'a'];
if (sCnt == pCnt)
ans.emplace_back(i + 1);
}
return ans;
}
};
法二:滑动窗口优化
参考官方题解
class Solution {
public:
vector<int> findAnagrams(string s, string p) {
if (p.size() > s.size())
return vector<int>();
int pLen = p.size(), differ = 0;
vector<int> ans;
vector<int> cnt(26);
for (int i = 0; i < pLen; i++)
{
//注意是在s 中找字母异位词
++cnt[s[i] - 'a'];
--cnt[p[i] - 'a'];
}
for (int i = 0; i < 26; i++)
{
if (cnt[i] != 0)
differ++;
}
if (differ == 0)
ans.emplace_back(0);
for (int i = 0; i < s.size() - pLen; i++)
{
//窗口中字母 s[i]的数量 与 字符串p中对应的数量由 相等 变为 不相等
if (cnt[s[i] - 'a'] == 0)
differ++;
else if (cnt[s[i] - 'a'] == 1)//窗口中字母 s[i]的数量 与 字符串p中对应的数量由 不相等 变为 相等
differ--;
--cnt[s[i] - 'a'];
if (cnt[s[i + pLen] - 'a'] == -1)//窗口中字母 s[i+pLen]的数量 与 字符串p中对应的数量由 不相等 变为 相等
differ--;
else if (cnt[s[i + pLen] - 'a'] == 0)//窗口中字母 s[i+pLen]的数量 与 字符串p中对应的数量由 相等 变为 不相等
differ++;
++cnt[s[i + pLen] - 'a'];
if (differ == 0)
ans.emplace_back(i + 1);
}
return ans;
}
};