438. 找到字符串中所有字母异位词

这是我参与11月更文挑战的第26天，活动详情查看：2021最后一次更文挑战

438. 找到字符串中所有字母异位词

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

解题思路

若想快速判断两个字符串是否为异位词，只需要判断两个字符串的字符组成是否相同，因此使用数组表示 s 中子串的组成以及字符串p的组成，如果表示二者组成数组如果完全相同，则代表二者是异位词。

使用滑动窗口，维护一个大小为p.size()的窗口，每次移动该窗口代表选取下一个子串，因此需要修改一下窗口内子串的组成，检查每个窗口和字符串p的组成数组是否相同，如果相同的话，代表当前子串为字符串p的一个异位词。

代码

class Solution {
public:
    vector<int> findAnagrams(string s, string p) {
        int a[26]={0},b[26]={0};
        for (int i = 0; i < p.size(); ++i) {
            a[p[i]-'a']++;
        }
        int l=0,r=0;
        vector<int> res;
        while (r<s.size()){
            if (r-l>p.size()){
                b[s[l]-'a']--;
                l++;
            }
            if (is_same(a,b))
                res.push_back(l);
            b[s[r]-'a']++;
            r++;
        }
        if (r-l>p.size()){
            b[s[l]-'a']--;
            l++;
        }
        if (is_same(a,b))
            res.push_back(l);
        return res;

    }
    bool is_same(int a[26],int b[26]) {

        for (int i = 0; i < 26; ++i) {
            if (a[i]!=b[i])
                return false;
        }
        return true;
    }
};