[滑动窗口]找出最长等值子数组

题目

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;
    }
};

  • 14
    点赞
  • 10
    收藏
    觉得还不错? 一键收藏
  • 1
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值