力扣2200. 找出数组中的所有 K 近邻下标


这一题的大意是说让我们找到一个下标,与它相邻k个单位内的元素中,有没有一个下标对应的值等于key,如果有,那么就说明这个下标符合条件,把下标放到ans数组中,最后返回ans数组。

这一题的数据范围是10^3,可以轻松的暴力通过,难点在于如何用滑动窗口实现O(n),这一题应该也可以用差分加前缀和来实现O(n)的

下面我们来看一下怎么用滑动窗口的方式实现O(n)

首先我们应该找到第一个窗口中,对应的等于key的元素的下标是多少,用last表示,

然后,我们就可以遍历所有的元素,看在该元素所处的滑动窗口中,是否有等于key的元素存在,如果不存在,如果存在我们就可以把遍历到的元素放入到ans数组中,

要注意的是第一个滑动窗口的大小是k,但窗口最大会等于2*k-1,因此,在找是否窗口中有符合条件的下标,只需要last>=i-k即可(大于窗口的左边界)而不是last<=i+k(小于窗口的右边界),为啥不能是小于窗口的右边界呢?(这也是我之前没有AC的原因)因为可能在窗口的移动过程中,某一个窗口中并没有符合条件的下标,那么last就无法被更小,此时last可能是之前的窗口中的下标,如果用last<=i+k(小于窗口的右边界)来判断是否在窗口中有符合条件的下标,就会导致出错,last可能是之前的窗口中的下标。

因此代码如下:

cpp 复制代码
class Solution {
public:
    vector<int> findKDistantIndices(vector<int>& nums, int key, int k) {
        vector<int> ans;
        int n=nums.size();
        int last=0-k-1;
      for(int i=k-1;i>=0;i--)
      {
        if(nums[i]==key)
        {
            last=i;
            break;
        }
      }
      for(int i=0;i<n;i++)
      {
         if(i+k<nums.size()&&nums[i+k]==key)
          {
            last=i+k;
          }
          if(last>=i-k)
          {
              ans.push_back(i);
          }
          
         
      }
       
        return ans;
    }
};

时间复杂度O(n)

相关推荐
大熊背32 分钟前
树莓派相机自动白平衡详解(四)
算法·树莓派·白平衡·isppipeline
一米阳光866135 分钟前
软考(中级)软件设计师核心笔记(9)算法——背包问题
笔记·算法·职场发展·软考·软件设计师·中级职称
晚风醉蝶1 小时前
1-11-奇偶排序-OddEvenSort
java·数据结构·算法
旖旎夜光1 小时前
LeetCode 852:山脉数组的峰顶索引(二分查找) —— 题解
数据结构·c++·算法·leetcode·二分查找
白露与泡影3 小时前
解密 Pi 的 Harness 工程:Agent 会话如何实现持久化与恢复
java·人工智能·算法
happyprince4 小时前
01-整体观-Codex全貌解读(源码)
算法·ai编程
俊基科技5 小时前
别自己写算法了,让这颗37.5毫米的模组替你“闭嘴干活”
神经网络·算法·硬件开发·ai降噪·语音模块·回音消除
_Narcissus_5 小时前
动态规划初探(含完整代码实现)
c语言·数据结构·c++·算法·动态规划·背包问题·最短路径
世人万千丶6 小时前
鸿蒙项目实战 - 社区活动编排板:标签云布局算法与自动换行
学习·算法·华为·harmonyos·鸿蒙