leetcode 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" 的异位词

//统计字母数量进行比对而非sort
//滑动窗口每次去掉一个最前面的元素，加入一个下一个新元素
//每次去掉加入都需重新sort，时间复杂度O(nlogn)
//而统计字母数量则仅需2次计算，时间复杂度O(1)
class Solution {
public:

    vector<int> findAnagrams(string s, string p) {
        vector<int> res;
        vector<int> sCount(26);
        vector<int> pCount(26);

        if(p.size()>s.size()) return res;
        for(int i=0,j=0;i<p.size();i++){
            sCount[s[i]-'a']++;
            pCount[p[i]-'a']++;
        }
        if(sCount==pCount) res.push_back(0);

        for(int i=1;i<s.size();i++){
            if(i+p.size()-1>=s.size()) break;
            sCount[s[i-1]-'a']--;
            sCount[s[i+p.size()-1]-'a']++;
            if(sCount==pCount) res.push_back(i);
        }
        return res;
    }

};