LeetCode 438. 找到字符串中所有字母异位词 ---滑动窗口+区间字母数目标记

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 仅包含小写字母

题解

这个题目比较灵活，首先滑动窗口是需要的，这个不用说，然后考虑下某个窗口内的内容可以构成p字符串的内容，我们首先考虑到数目关系，如果滑动窗口滑动到某个区间内的所有字符的数目和p的所有字符数目一致，那么就是一个答案。

AC代码

class Solution {
public:
    int s_num[26],p_num[26];
    vector<int>res;
    vector<int> findAnagrams(string s, string p) 
    {
        if(s.length()<p.length())return res;//特判
        memset(s_num,0,sizeof(s_num));
        memset(p_num,0,sizeof(p_num));
        for(int i=0;i<p.length();i++)
        {
            p_num[p[i]-'a'] ++;
        }
        int ans = 0;//表示窗口中有多少个字母的数目满足了p中的要求
        //滑动窗口
        for(int i=0;i<p.length();i++)
        {
            s_num[s[i]-'a']++;
        }
        for(int i=0;i<26;i++)
        {
            if(s_num[i] == p_num[i])ans++;
        }
        if(ans == 26)
        res.push_back(0);
        for(int i=p.length();i<s.length();i++)
        {
            int pre = i-p.length();
            if(s_num[s[pre]-'a']==p_num[s[pre]-'a'])
            ans --;
            s_num[s[pre]-'a']--;
            if(s_num[s[pre]-'a'] == p_num[s[pre]-'a'])
            ans ++;
            if(s_num[s[i]-'a']==p_num[s[i]-'a'])
            ans --;
            s_num[s[i]-'a'] ++;
            if(s_num[s[i]-'a']==p_num[s[i]-'a'])
            ans ++;
            if(ans == 26)
            res.push_back(pre+1);
        }
        return res;
    }
};