【LeetCode热题100】【滑动窗口】找到字符串中所有字母异位词

Ye-Maolin2023-12-10 22:44

给定两个字符串 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 仅包含小写字母

题解

一开始是想用两层循环，先将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;
    }
};