当前位置：首页 > news >正文

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

news 2025/10/5 6:42:39

在这里插入图片描述

这一题的大意是说让我们找到一个下标，与它相邻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可能是之前的窗口中的下标。
因此代码如下：

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