【LeeCode】找到字符串中所有字母异位词

给定两个字符串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” 的异位词。

核心思路：

滑动窗口的基本思路是维护一个长度为p的窗口，在字符串s中滑动这个窗口，并检查窗口中的子串是否为字符串p的异位词。异位词本质就是出现字母的频率是相同的。

#include <iostream>
#include <vector>
#include <string>

using namespace std;

class Solution 
{
public:
    vector<int> findAnagrams(string s, string p) 
    {
        vector<int> result;
        if (s.size() < p.size())
            return result;
        //p_count记录p中各字符的频率，s_count用来记录每个窗口变化的字符频率
        vector<int> p_count(26, 0), s_count(26, 0);
        for (int i = 0; i < p.size(); i++)
        {
            p_count[p[i] - 'a']++;
            s_count[s[i] - 'a']++;
        }
        // 滑动窗口
        for (int i = p.size(); i < s.size(); i++)
        {
            if (p_count == s_count)
                result.push_back(i - p.size());
            // 更新窗口
            s_count[s[i] - 'a']++;
            // 移除旧的元素
            s_count[s[i - p.size()] - 'a']--;
        }
        // 检查最后最后一个窗口
        if (p_count == s_count)
            result.push_back(s.size() - p.size());

        return result;
    }
};

int main() 
{
    Solution solution;
    string s = "cbaebabacd";
    string p = "abc";
    vector<int> result = solution.findAnagrams(s, p);

    cout << "Anagram indices: ";
    for (int index : result) 
    {
        cout << index << " ";
    }
    cout << endl;

    return 0;
}