2831.找出最长等值子数组(哈希表+滑动窗口法)

给你一个下标从 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.元素分类: 构建一个哈希表,用来记录数字中所有出现的元素以及对应的位置
2.设置窗口: 对每个元素,采用滑动窗口去寻找最长的子数组,窗口的左右边界为[i,j]
3.滑动窗口: 在滑动窗口时,要保证窗口内删除的元素不能超过k,超过则把ij移动
4.计算窗口长度: 每滑动一次窗口,就计算一次窗口长度:i-j+1,并和全局长度进行比较,保留较长的

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 &[_,vec]:pos){
            int i=0;//窗口左端点
            for(int j=0;j<vec.size();j++){//窗口右端点,不停的向右滑动
                while(vec[j]-vec[i]-(j-i)>k){//每滑动一次就判断一次
                    i++;
                }
                res=max(res,j-i+1);//更新全局最大长度
            }
        }
        return res;
    }
};

题后学习

假设我们的 unordered_map pos 如下所示:

unordered_map<int, vector<int>> pos = {
    {1, {0, 2, 4}},
    {2, {1, 3}},
    {3, {5}}
};

在这个哈希表中:

键 1 对应的值是一个向量 {0, 2, 4}。
键 2 对应的值是一个向量 {1, 3}。
键 3 对应的值是一个向量 {5}。

嵌套循环代码

for(auto &[_,vec]:pos){
    int i=0; // 窗口左端点
    for(int j=0;j<vec.size();j++)
        // ...
  1. 第一次迭代(外层循环处理键 1):
    vec 引用向量 {0, 2, 4}
    i 初始化为0
    内层循环将遍历 {0, 2, 4},j 从0开始,依次为0, 1, 2。

  2. 第二次迭代(外层循环处理键 2):
    vec 引用向量 {1, 3}
    i 再次初始化为0
    内层循环将遍历 {1, 3},j 从0开始,依次为0, 1。

  3. 第三次迭代(外层循环处理键 3):
    vec 引用向量 {5}
    i 再次初始化为0
    内层循环将遍历 {5},j 从0开始,但因为只有一个元素,所以只执行一次。

  • 5
    点赞
  • 11
    收藏
    觉得还不错? 一键收藏
  • 打赏
    打赏
  • 0
    评论

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

青石横刀策马

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

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

余额充值