leetcode 438. 找到字符串中所有字母异位词

@(labuladong的算法小抄)[滑动窗口]

leetcode 438. 找到字符串中所有字母异位词

题目描述

解题思路

参考：labuladong的算法小抄P93

class Solution {
    public List<Integer> findAnagrams(String s, String p) {
        /* needs记录需要凑齐的各字符的总数量 */
        Map<Character, Integer> needs = new HashMap<>();
        /* cnt记录窗口中已凑的各字符的数量 */
        Map<Character, Integer> cnt = new HashMap<>();
        /* 初始化needs,相当于c++的 needs[c]++ */
        for (char c : p.toCharArray())
            needs.put(c, needs.getOrDefault(c, 0) + 1);
        /* 窗口的两端，注意是左闭右开[left, right)，这样窗口的大小就直接是right-left */
        int left = 0, right = 0;
        /* 表示cnt中满足needs对应的字符的个数 */
        int valid = 0;
        List<Integer> res = new LinkedList<>();

        while (right < s.length()) {
            /* intoWindow是移入窗口的字符，同时也是可能移入cnt的字符 */
            char intoWindow = s.charAt(right);
            /* 扩大右侧窗口 */
            right++;
            /* 如果该字符是needs中需要凑的字符 */
            if (needs.containsKey(intoWindow)) {
                /* cnt[intoWindow]++ */
                cnt.put(intoWindow, cnt.getOrDefault(intoWindow, 0) + 1);
                /* 如果该字符凑齐了，则valid++ */
                if (needs.get(intoWindow).equals(cnt.get(intoWindow)))
                    valid++;
            }
            /* 判断左侧窗口是否需要收缩，只要窗口大小达到p的长度就进行判断,相当于一个p长度大小的窗口在逐个往右滑 */
            while (right - left >= p.length()) {
                /* 判断是否找到合法子串 */
                if (valid == needs.size()) {
                    res.add(left);
                }
                /* outOfWindow是移出窗口的字符，同时也是可能移出cnt的字符 */
                char outOfWindow = s.charAt(left);
                /* 收缩左侧窗口 */
                left++;
                /* 如果该字符是needs中需要凑的字符 */
                if (needs.containsKey(outOfWindow)) {
                    /* 如果该字符在移出前刚好处于凑齐的状态，移出后正好不满足needs条件，则valid-- */
                    if (needs.get(outOfWindow).equals(cnt.get(outOfWindow)))
                        valid--;
                    /* cnt[outOfWindow]-- */
                    cnt.put(outOfWindow, cnt.getOrDefault(outOfWindow, 0) - 1);
                }
            }
        }
        return res;
    }
}