Find All K-Distant Indices in an Array
给一个数组和一个key和一个数字k, 求数组中等于k的index的范围k的所有index, 要求答案排序.
用linkedhashmap去重.
class Solution {
public List<Integer> findKDistantIndices(int[] nums, int key, int k) {
List<Integer> res = new ArrayList<>();
Set<Integer> set = new LinkedHashSet<>();
for(int i = 0; i < nums.length; i++) {
if(nums[i] == key) {
for(int j = Math.max(i - k, 0); j <= Math.min(i + k, nums.length - 1); j++)
set.add(j);
}
}
for(int n : set)
res.add(n);
return res;
}
}