LeetCode 1852. 每个子数组的数字种类数

1852. 每个子数组的数字种类数

给你一个整数数组 nums与一个整数 k，请你构造一个长度 n-k+1 的数组 ans，这个数组第i个元素 ans[i] 是每个长度为k的子数组 nums[i:i+k-1] = [nums[i], nums[i+1], ..., nums[i+k-1]]中数字的种类数。

返回这个数组 ans。

示例 1:

输入: nums = [1,2,3,2,2,1,3], k = 3
输出: [3,2,2,2,3]
解释：每个子数组的数字种类计算方法如下：
- nums[0:2] = [1,2,3] 所以 ans[0] = 3
- nums[1:3] = [2,3,2] 所以 ans[1] = 2
- nums[2:4] = [3,2,2] 所以 ans[2] = 2
- nums[3:5] = [2,2,1] 所以 ans[3] = 2
- nums[4:6] = [2,1,3] 所以 ans[4] = 3

示例 2:

输入: nums = [1,1,1,1,2,3,4], k = 4
输出: [1,2,3,4]
解释: 每个子数组的数字种类计算方法如下：
- nums[0:3] = [1,1,1,1] 所以 ans[0] = 1
- nums[1:4] = [1,1,1,2] 所以 ans[1] = 2
- nums[2:5] = [1,1,2,3] 所以 ans[2] = 3
- nums[3:6] = [1,2,3,4] 所以 ans[3] = 4

提示:

• 1 <= k <= nums.length <= 10^5
• 1 <= nums[i] <= 10^5

---

提示 1

Keep a frequency map of the elements in each window.

---

提示 2

When the frequency of the element is 0, remove it from the map.

---

提示 3

The answer to each window is the size of the map.

 

 解法：滑动窗口 (定长、单指针)

class Solution {
    public int[] distinctNumbers(int[] nums, int k) {
        int n = nums.length;
        int[] ans = new int[n - k + 1];
        Map<Integer, Integer> window = new HashMap<>();
        int i = 0;
        for (; i < k - 1; i++) {
            window.put(nums[i], window.getOrDefault(nums[i], 0) + 1);
        }
        // 开始时 i == k - 1
        while (i < n) {
            window.put(nums[i], window.getOrDefault(nums[i], 0) + 1);
            ans[i - k + 1] = window.size();
            window.put(nums[i - k + 1], window.get(nums[i - k + 1]) - 1);
            if (window.get(nums[i - k + 1]) == 0) {
                window.remove(nums[i - k + 1]);
            }
            i++;
        }
        return ans;
    }
}

 执行用时 73ms   消耗内存 59.6MB

复杂度分析

• 时间复杂度：O(n)，n 是 数组的长度。
• 空间复杂度：O(n)。

 另一种写法：

class Solution {
    public int[] distinctNumbers(int[] nums, int k) {
        int n = nums.length;
        int[] ans = new int[n - k + 1];
        Map<Integer, Integer> window = new HashMap<>();
        int i = 0;
        for (; i < k - 1; i++) {
            window.merge(nums[i], 1, Integer::sum);
        }
        // 开始时 i == k - 1
        while (i < n) {
            window.merge(nums[i], 1, Integer::sum);
            ans[i - k + 1] = window.size();
            if (window.merge(nums[i - k + 1], -1, Integer::sum) == 0) {
                window.remove(nums[i - k + 1]);
            }
            i++;
        }
        return ans;
    }
}

 执行用时 59ms   消耗内存 58.87MB

复杂度分析

• 时间复杂度：O(n)，n 是 数组的长度。
• 空间复杂度：O(n)。