LeetCode 438. Find All Anagrams in a String

LeetCode 438. Find All Anagrams in a String (找到字符串中所有字母异位词)

题目

链接

https://leetcode-cn.com/problems/find-all-anagrams-in-a-string/

问题描述

给定两个字符串 s 和 p，找到 s 中所有 p 的 异位词 的子串，返回这些子串的起始索引。不考虑答案输出的顺序。

异位词 指由相同字母重排列形成的字符串（包括相同的字符串）。

示例

输入: s = "cbaebabacd", p = "abc"
输出: [0,6]
解释:
起始索引等于 0 的子串是 "cba", 它是 "abc" 的异位词。
起始索引等于 6 的子串是 "bac", 它是 "abc" 的异位词。

提示

1 <= s.length, p.length <= 3 * 104
s 和 p 仅包含小写字母

思路

同样是滑动窗口，只需要用window和need两个hashmap存储，leftright进行窗口滑动。对于符合条件的数存放到list当中，最后返回即可。

复杂度分析

时间复杂度 O(n)
空间复杂度 O(n)

代码

Java

    public static List<Integer> findAnagrams(String s, String p) {
        List<Integer> ans = new ArrayList<>();
        Map<Character, Integer> window = new HashMap();
        Map<Character, Integer> need = new HashMap();
        int plen = p.length();
        int slen = s.length();
        for (int i = 0; i < plen; i++) {
            char c = p.charAt(i);
            need.put(c, need.getOrDefault(c, 0) + 1);
        }
        int left = 0, right = 0;
        int valid = 0;
        while (right < slen) {
            char addch = s.charAt(right);
            window.put(addch, window.getOrDefault(addch, 0) + 1);
            right++;
            if (need.containsKey(addch) && window.get(addch).equals(need.get(addch))) {
                valid++;
                if (valid == need.size()) {
                    ans.add(left);
                }
            }
            while (right - left >= plen) {
                char rech = s.charAt(left);
                left++;
                if (need.containsKey(rech)) {

                    if (window.get(rech).equals(need.get(rech))) {
                        valid--;
                    }window.put(rech, window.get(rech) - 1);
                }
            }
        }
        return ans;
    }