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 * 104
• s 和 p 仅包含小写字母

解答

class Solution {
public:
    vector<int> findAnagrams(string s, string p) {
        // 方法：滑动窗口
        // 用大小为26的数组表示字符串中【字母出现次数】
        // 先建立窗口，大小为p.size()
        // 随后滑动窗口进行判断
        int ns = s.size(), np = p.size();
        if(ns < np) return {};

        vector<int> res;
        vector<int> pCnt(26, 0);
        vector<int> sCnt(26, 0);
    
        // 1.建立窗口
        // 建立长度为np的窗口进行滑动
        for(int i = 0; i < np; ++i)
        {
            pCnt[p[i] - 'a']++; // 记录p串每个字母出现次数
            sCnt[s[i] - 'a']++;
        }
        if(pCnt == sCnt) res.push_back(0);

        // 2. 滑动窗口
        for(int i = np; i < ns; ++i) // i表示窗口新的右边界
        {
            sCnt[s[i - np] - 'a']--; // 窗口左边界右移
            sCnt[s[i] - 'a']++;

            if(pCnt == sCnt) res.push_back(i- np + 1); 
        }
        return res;
    }
};