image_2022-03-18_14-52-30.png
43.1 KB
#N2200. Find All K-Distant Indices in an Array
problem link
#solution
problem link
#solution
class Solution {
public List<Integer> findKDistantIndices(int[] nums, int key, int k) {
Set<Integer> set = new TreeSet<>();
int l=nums.length;
for(int i=0; i<l; i++){
if(nums[i]==key){
set.add(i);
for(int j=i-k; j<=i+k; j++){
if(j>-1 && j<l)
set.add(j);
}
}
}
return new ArrayList<Integer>(set);
}
}👍1