给你一个下标从 0 开始的整数数组 nums 和一个整数 k 。
如果子数组中所有元素都相等,则认为子数组是一个 等值子数组 。注意,空数组是 等值子数组 。
从 nums 中删除最多 k 个元素后,返回可能的最长等值子数组的长度。
子数组 是数组中一个连续且可能为空的元素序列。
示例 1:
输入:nums = [1,3,2,3,1,3], k = 3
输出:3
解释:最优的方案是删除下标 2 和下标 4 的元素。
删除后,nums 等于 [1, 3, 3, 3] 。
最长等值子数组从 i = 1 开始到 j = 3 结束,长度等于 3 。
可以证明无法创建更长的等值子数组。
示例 2:
输入:nums = [1,1,2,2,1,1], k = 2
输出:4
解释:最优的方案是删除下标 2 和下标 3 的元素。
删除后,nums 等于 [1, 1, 1, 1] 。
数组自身就是等值子数组,长度等于 4 。
可以证明无法创建更长的等值子数组。
提示:
1 <= nums.length <= 105
1 <= nums[i] <= nums.length
0 <= k <= nums.length
**思路:i 和 j 是滑动窗口的两个指针,分别表示当前窗口的右边界和左边界。
v[i] - v[j] 表示窗口的实际跨度,而 i - j 表示窗口内的元素数量。
v[i] - v[j] - (i - j) 表示在窗口跨度内,必须删除的元素数量,如果这个数量超过了 k,则窗口左边界 j 右移。
**
class Solution {
public:
int longestEqualSubarray(vector<int>& nums, int k) {
int n = nums.size();
unordered_map<int, vector<int>> pos;
for(int i = 0; i < n; i ++ ) {
pos[nums[i]].push_back(i);
}
int res = 0;
for(auto [key, v] : pos) {
for(int i = 0, j = 0; i < v.size(); i ++ ) {
while(v[i] - v[j] - (i - j) > k) {
j ++;
}
res = max(res, i - j + 1);
}
}
return res;
}
};