题目描述
给定一个整数数组和一个整数 k,判断数组中是否存在两个不同的索引 i 和 j,使得 nums [i] = nums [j],并且 i 和 j 的差的绝对值最大为 k。
示例
输入:nums = [1, 2, 3, 1], k = 3
输出:true
输入:nums = [1, 0, 1, 1], k = 1
输出:true
输入:nums = [1, 2, 3, 1, 2, 3], k = 2
输出:false
解题思路
在本题中,首先要做的是找到两个相同的数,然后比较其索引之差,是否大于k,是则返回false,否则返回true。
如果从首元素开始遍历会出现一个问题,当找到相同的两个数索引之差大于k,但在未遍历的元素还会存在相同的元素,这样就不能简单的直接返回了。
所以我们选择散列表这一数据结构,将元素本身存储为key,其索引存储为value,如果散列表中无此key时,直接存储,当存在该key时,判断索引之差是否大于k,是则将之前索引替换为当前索引继续遍历,否则直接返回true,最后循环外直接返回false,因为没有符合要求的元素。
代码
public boolean containsNearbyDuplicate(int[] nums, int k) {
if (k < 1 || nums == null || nums.length < 2) return false;
HashMap<Integer, Integer> hashMap = new HashMap<>();
for (int i = 0; i < nums.length; i++) {
if (!hashMap.containsKey(nums[i])) hashMap.put(nums[i], i);
else {
int temp = hashMap.get(nums[i]);
if (i - temp <= k) return true;
hashMap.put(nums[i], i);
}
}
return false;
}
执行结果
欢迎各位大神指点