给你一个下标从 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 。 可以证明无法创建更长的等值子数组。
cpp
class Solution {
public:
int longestEqualSubarray(vector<int>& nums, int k) {
int left = 0, right = 0, n = nums.size();
int res = 0; // 某个出现最大的次数
unordered_map<int, int> mp;
while(right < n)
{
mp[nums[right]]++; // 进窗口
res = max(res, mp[nums[right]]); // 维护这个res记录出现次数最多的值
if(right-left+1 - res > k) // 只有当需要删除的元素大于这个k时需要出窗口
mp[nums[left++]]--; // 出窗口
right++;
}
return res;
}
};