问题描述:
Given an integer array nums and an integer k, return true if there are two distinct indices i and j in the array such that nums[i] == nums[j] and abs(i - j) <= k.
找在一定间距内相同的元素
思路:用hashmap,记录元素及其出现的位置。遍历数组,如果遇到了表里没有的元素,就把它加进去;如果表里有,我们判断间距,如果在间距内,则返回true,否则替换掉原来的元素。
代码:
class Solution {
public boolean containsNearbyDuplicate(int[] nums, int k) {
Map<Integer, Integer> map = new HashMap<>();
for (int i=0; i<nums.length; i++){
if (!map.containsKey(nums[i])){
map.put(nums[i], i);
}
else{
if ((i-map.get(nums[i]))<=k){
return true;
}
else{
map.put(nums[i], i);
}
}
}
return false;
}
}
方法二: 如果想节省空间也可以使用hashset。把set内的容量始终控制为k,如果目前在判断的元素在set里有,就返回true,如果没有就把它加入。但若这时容量超过k,就可以把第[i-k]个元素删除,因为他离我们的距离超过了k,也就不会影响运行结果了。
时间复杂度: O(n)