题目地址:
https://www.lintcode.com/problem/contains-duplicate-ii/description
给定一个数组 A A A,再给定一个整数 k k k,问是否有两个不同位置的相同数的下标之差的绝对值小于等于 k k k。
用哈希表存每个数出现的最后一次位置,当遍历到 A [ i ] A[i] A[i]的时候先看一下之前是否出现过,出现过则看一下下标差是否小于等于了 k k k。代码如下:
import java.util.HashMap;
import java.util.Map;
public class Solution {
/**
* @param nums: the given array
* @param k: the given number
* @return: whether there are two distinct indices i and j in the array such that nums[i] = nums[j] and the absolute difference between i and j is at most k
*/
public boolean containsNearbyDuplicate(int[] nums, int k) {
// Write your code here
Map<Integer, Integer> map = new HashMap<>();
for (int i = 0; i < nums.length; i++) {
if (map.containsKey(nums[i]) && i - map.get(nums[i]) <= k) {
return true;
}
map.put(nums[i], i);
}
return false;
}
}
时空复杂度 O ( n ) O(n) O(n)。