题目描述:
给定一个字符串 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" 的字母异位词。
思路:
1、定义两个相同类型的容器A,B,其中一个A存储字符串p中出现的字母以及出现的次数
2、开始遍历字符串s,当字符串中的字符出现在A中,则在B中添加,如果B中已存在,则增加B中该字符的计数值
3、当遍历长度达到p的长度时,判断A,B是否相等,相等则把起始位置保存,此时的起始位置为0,遍历达到的位置为x
4、继续遍历,则起始位置递增为1,遍历达到位置递增为x+1。此时需要将0位置的字符从B中拿出,x+1位置的字符放入B中,再与A进行比较。
代码:
class Solution {
public:
vector<int> findAnagrams(string s, string p) {
unordered_map<char,int>m,n;
vector<int>res;
int len=p.size(),head_index=0;
for(auto i : p)
m[i]++;
for(int i=0;i<s.size();i++)
{
if(m.find(s[i])!=m.end())
n[s[i]]++;
if(i<len-1)continue;
if(n==m)res.push_back(head_index);
if(n.find(s[head_index])!=n.end())
n[s[head_index]]--;
head_index++;
}
return res;
}
};
改进:
class Solution {
public:
vector<int> findAnagrams(string s, string p) {
vector<int>res, n(26,0),m(26,0);
int len=p.size(),head_index=0;
for(auto i : p)
m[i-'a']++;
for(int i=0;i<s.size();i++)
{
n[s[i]-'a']++;
if(i<len-1)continue;
if(n==m)res.push_back(head_index);
n[s[head_index]-'a']--;
head_index++;
}
return res;
}
};