题目:
2831. 找出最长等值子数组 - 力扣(LeetCode)
给你一个下标从 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 。 可以证明无法创建更长的等值子数组。
解题思路
题目实际上是让找一段最长的区间,这个区间内的首(i)和尾(j)的下标对应的值相等,并且中间不等于首尾的值的个数应该小于等于k,既然是让找区间并且根据题目的性质首先就会想到滑动窗口。
首先我们需要一个新的数组check来记录这个窗口中每个值出现的次数,left是窗口的左边界,right是窗口的右边界,还需要一个变量ret来存窗口的最大值。
进窗口
将right指向的nums中的元素的个数放到check数组中。
出窗口
先看什么时候需要出窗口,我们以left为基准,每次检查在这个窗口中的除了以left为下标对应的元素的个数(也就是找以left为基准的等值数组),与k的值进行比较,如果小于等于k的话就说明是足以将除nums[left]的值删去,当大于k的话就说明不够删了,这时候就需要出窗口了,注意这里是一次一次走的,所以出窗口的时候仅需让left向右移动一次仅可,同时将left对应元素的个数减一。
判断
当判断完出窗口之后,此时的窗口就是符合条件了,就可以进行判断,但是这里如果判段left所存元素的个数就判断不全,所以判断的条件是判断right所存元素的个数,但是为什么判断right是可以的?
此时不外乎两种情况:nums[left]==nums[right],这种情况在出窗口的时候就已经判断过了,是符合条件的。另一种情况就是 nums[left]!=nums[right],这种情况我们可以将这个窗口中的值看成三种元素,A=num[left],B=nums[right],C是剩下的元素,这里也需要分两种情况:假设B的首尾之间不等于A的个数大于k的话,这显然是不可能的,因为出窗口的时候判断了除left所在的元素是小于k的。那么另一种情况就是B中首尾之间的有A和C(C是绝对小于k的)的个数是大于k,此时的B的个数会被计入在内,但是计入在内并不会受影响,因为这种情况计入的个数是小于之前存的个数的,对结果并不会产生影响。
代码
class Solution
{
public:
int longestEqualSubarray(vector<int>& nums, int k)
{
int ret = 0,n = nums.size();
vector<int> check(n+1,0);
for(int left = 0,right = 0;right<n;right++)
{
check[nums[right]]++;
if(right-left+1-check[nums[left]]>k) check[nums[left++]]--;
ret = max(ret,check[nums[right]]);
}
return ret;
}
};