给你一个整数数组 nums
和一个整数 k
,判断数组中是否存在两个 不同的索引 i
和 j
,满足 nums[i] == nums[j]
且 abs(i - j) <= k
。如果存在,返回 true
;否则,返回 false
。
示例 1:
输入:nums = [1,2,3,1], k = 3 输出:true
示例 2:
输入:nums = [1,0,1,1], k = 1 输出:true
示例 3:
输入:nums = [1,2,3,1,2,3], k = 2 输出:false
提示:
1 <= nums.length <= 105
-109 <= nums[i] <= 109
0 <= k <= 105
方法一:哈希表
我们用哈希表存放最近遍历到的数以及对应的下标。
遍历数组 nums
,对于当前遍历到的元素 true
,否则将当前元素加入哈希表中。
遍历结束后,返回 false
。
时间复杂度 nums
的长度。
class Solution:
def containsNearbyDuplicate(self, nums: List[int], k: int) -> bool:
mp = {}
for i, v in enumerate(nums):
if v in mp and i - mp[v] <= k:
return True
mp[v] = i
return False
class Solution {
public boolean containsNearbyDuplicate(int[] nums, int k) {
Map<Integer, Integer> mp = new HashMap<>();
for (int i = 0; i < nums.length; ++i) {
if (mp.containsKey(nums[i]) && i - mp.get(nums[i]) <= k) {
return true;
}
mp.put(nums[i], i);
}
return false;
}
}
class Solution {
public:
bool containsNearbyDuplicate(vector<int>& nums, int k) {
unordered_map<int, int> mp;
for (int i = 0; i < nums.size(); ++i) {
if (mp.count(nums[i]) && i - mp[nums[i]] <= k) return true;
mp[nums[i]] = i;
}
return false;
}
};
func containsNearbyDuplicate(nums []int, k int) bool {
mp := make(map[int]int)
for i, v := range nums {
if j, ok := mp[v]; ok {
if i-j <= k {
return true
}
}
mp[v] = i
}
return false
}
public class Solution {
public bool ContainsNearbyDuplicate(int[] nums, int k) {
var mp = new Dictionary<int, int>();
for (int i = 0; i < nums.Length; ++i)
{
if (mp.ContainsKey(nums[i]) && i - mp[nums[i]] <= k)
{
return true;
}
mp[nums[i]] = i;
}
return false;
}
}