LeetCode 1852. 每个子数组的数字种类数（滑窗）

文章目录

- 1. 题目
- 2. 解题

1. 题目

给你一个整数数组 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] 有'1','2','3'三种数字所以      ans[0] = 3
- nums[1:3] = [2,3,2] 有'2','3'两种数字所以          ans[1] = 2
- nums[2:4] = [3,2,2] 有'2','3'两种数字所以          ans[2] = 2
- nums[3:5] = [2,2,1] 有'1','2'两种数字所以          ans[3] = 2
- nums[4:6] = [2,1,3] 有'1','2','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] 只有'1'这一种数字所以         ans[0] = 1
- nums[1:4] = [1,1,1,2] 有'1','2'两种数字所以         ans[1] = 2
- nums[2:5] = [1,1,2,3] 有'1','2','3'三种数字所以     ans[2] = 3
- nums[3:6] = [1,2,3,4] 有'1','2','3','4'四种数字所以 ans[3] = 4提示:
1 <= k <= nums.length <= 10^5
1 <= nums[i] <= 10^5

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

2. 解题

滑动窗口

class Solution {
public:vector<int> distinctNumbers(vector<int>& nums, int k) {int n = nums.size();vector<int> ans(n-k+1);unordered_map<int,int> m;for(int i = 0; i < k; ++i){m[nums[i]]++;}ans[0] = m.size();for(int i = k; i < n; ++i){if(--m[nums[i-k]] == 0)m.erase(nums[i-k]);++m[nums[i]];ans[i-k+1] = m.size();}return ans;}
};