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

给定一个字符串 s 和一个非空字符串 p，找到 s 中所有是 p 的字母异位词的子串，返回这些子串的起始索引。

字符串只包含小写英文字母，并且字符串 s 和 p 的长度都不超过 20100。

说明：

字母异位词指字母相同，但排列不同的字符串。
不考虑答案输出的顺序。
示例 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" 的字母异位词。

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/find-all-anagrams-in-a-string
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

感谢labuladong大神，传送门我写了套框架，把滑动窗口算法变成了默写题。

class Solution {
        public List<Integer> findAnagrams(String s, String p) {
            List<Integer> res = new ArrayList<>();
            if (s == null || s.length() == 0) return res;
//            return findAnagramsI(s, p, res);
            return findAnagramsII(s, p, res);
        }

        //方法二：使用数组
        //时间复杂度O(mn)，空间复杂度O(1)
        private List<Integer> findAnagramsII(String s, String p, List<Integer> res) {
            int[] need = new int[256];
            int[] window = new int[256];

            for (char c : p.toCharArray()) {
                need[c]++;
            }
            int left = 0, right = 0;
            int valid = 0;
            char[] str = s.toCharArray();
            while (right < str.length) {
                char c = str[right];
                right++;
                window[c]++;

                //窗口中遇到字符串p的字符且满足p中字符出现的次数
                if (need[c] > 0 && need[c] >= window[c]) {
                    valid++;
                }

                while (right - left >= p.length()) {
                    if (valid == p.length()) {
                        res.add(left);
                    }

                    char d = str[left];
                    left++;

                    if (need[d] > 0 && need[d] >= window[d]) {
                        valid--;
                    }
                    window[d]--;
                }
            }
            return res;
        }

        //方法一：定义两个Map
        //时间复杂度O(mn)，空间复杂度O(m)，m为字符串p的长度
        private List<Integer> findAnagramsI(String s, String p, List<Integer> res) {
            Map<Character, Integer> need = new HashMap<>();
            Map<Character, Integer> window = new HashMap<>();
            for (char c : p.toCharArray()) {
                need.put(c, need.getOrDefault(c, 0) + 1);
            }
            int left = 0, right = 0;
            int valid = 0;
            char[] str = s.toCharArray();
            while (right < str.length) {
                //向右滑动窗口
                char c = str[right];
                right++;

                //更新窗口数据
                if (need.containsKey(c)) {
                    window.put(c, window.getOrDefault(c, 0) + 1);
                    //窗口中数据满足p中字符个数时
                    if (window.get(c).equals(need.get(c))) {
                        valid++;
                    }
                }

                //收缩窗口
                while (right - left >= p.length()) {
                    //更新结果
                    if (valid == need.size()) {
                        res.add(left);
                    }
                    //left向右移动，收缩窗口
                    char d = str[left];
                    left++;

                    //更新窗口数据
                    if (need.containsKey(d)) {
                        if (window.get(d).equals(need.get(d))) {
                            valid--;
                        }
                        window.put(d, window.get(d) - 1);
                    }
                }
            }
            return res;
        }
    }