题目链接:438. 找到字符串中所有字母异位词 - 力扣(LeetCode)
给定两个字符串 s
和 p
,找到 s
中所有 p
的 异位词 的子串,返回这些子串的起始索引。不考虑答案输出的顺序。
异位词 指由相同字母重排列形成的字符串(包括相同的字符串)。
一开始是想用两层循环,先将p排序一次,然后将s中每个和p一样长的子串拿出来重新排序之后和p比较是否相同,但是这种做法会超时
于是采用了官方的解法,官方的做法有两个优点,一个是利用了滑动窗口,另一个是将判断异位词转换成判断每个字母出现的次数是否相同,这个确实是最快判断是否是由相同字母组成的字符串的方法
class Solution {
public:
vector<int> findAnagrams(string s, string p) {
vector<int>answer;
int sLength=s.size(),pLength=p.size();
if(sLength<pLength){ // 如果s短于p,后面无法放置窗口
return {};
}
vector<int>ss(26),pp(26); // 记录字母出现次数
for(int i=0;i<pLength;i++){ // 放置滑动窗口
ss[s[i]-'a']++;
pp[p[i]-'a']++;
}
if(ss==pp)
answer.emplace_back(0);
for(int i=0;i<sLength-pLength;i++){
ss[s[i]-'a']--; // 滑动窗口移动,去掉前一个字母的状态
ss[s[i+pLength]-'a']++; // 滑动窗口移动,增加后一个字母的状态
if(ss==pp)
answer.emplace_back(i+1);
}
return answer;
}
};
rust
impl Solution {
pub fn find_anagrams(s: String, p: String) -> Vec<i32> {
if s.len()<p.len(){
return vec![];
}
let mut ans:Vec<i32>=vec![];
let mut target=[0;26];
let mut suspect=[0;26];
let offset=p.len();
let p=p.as_bytes();
let s=s.as_bytes();
for i in 0..offset{
target[(p[i]-b'a')as usize]+=1;
suspect[(s[i]-b'a')as usize]+=1;
}
if target==suspect{
ans.push(0);
}
for i in 0..(s.len()-offset){
suspect[(s[i]-b'a')as usize]-=1;
suspect[(s[i+offset]-b'a')as usize]+=1;
if target==suspect{
ans.push((i+1) as i32);
}
}
ans
}
}