题目描述(中等)
给定两个字符串 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 仅包含小写字母
思路
注意异位词的定义,由相同字母重排列形成的字符串(包括相同的字符串)
所以只需要统计每个字母出现的次数就好,如果两个字符串字母出现次数相同,则肯定满足异位词,并且长度必然相同
首先对长度判断,处理s比p短的情况,直接回空
然后统计p中的字频,同时统计s中0到n_p的字频(即第一段长度相同的字串)判断是否相等,相等添入0
之后用双指针逐位遍历s,每向后移一位,字频中去掉前指针之前的一位字母,加入后指针字母
每次判断是否相等,相等就在答案中加入前指针
最后返回答案即可
代码
class Solution {
public:
vector<int> findAnagrams(string s, string p) {
vector<int> ans;
int n_p = p.size();
int n_s = s.size();
if(n_s < n_p) return ans;
vector<int> p_count(27,0);//1-26存放字母数量,0不用
vector<int> tmp(27,0);
for(int i = 0; i < n_p; i++) {
p_count[p[i]-'a'+1]++;
tmp[s[i]-'a'+1]++;
}
if(tmp == p_count) ans.push_back(0);
for(int i = 1; i <= n_s - n_p; i++) {
int j = i + n_p - 1;
tmp[s[i-1]-'a'+1]--;
tmp[s[j]-'a'+1]++;
if(tmp == p_count) ans.push_back(i);
}
return ans;
}
};