[Leetcode] 219. Contains Duplicate II 解题报告

最新推荐文章于 2024-10-26 00:30:00 发布

原创最新推荐文章于 2024-10-26 00:30:00 发布 · 319 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#Leetcode #解题报告 #Hash

IT公司面试习题专栏收录该内容

736 篇文章

订阅专栏

本文介绍了一种使用哈希表来解决寻找数组中是否存在两个不同索引的相同元素且索引差不超过k的问题的方法。通过记录每个数出现的位置，并在发现重复数值时检查索引差是否符合条件，以此达到O(n)的时间复杂度。

题目：

Given an array of integers and an integer k, find out 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.

思路：

思想和 [Leetcode] 217 Contains Duplicate非常类型，只需要稍稍做一下改进即可：我们在哈希表中同时记录数组中每个数出现的位置。一旦发现了值相同的数，还需要判断两者的索引差距是否在k之内。如果是则直接返回true，否则我们还需要修改该数的索引（因为我们是从前往后扫描的，所以修改后的索引会有利于后续再次发现相同的数时对索引差异的判断）。算法的时间复杂度和空间复杂度都是O(n)。

代码：

class Solution {
public:
    bool containsNearbyDuplicate(vector<int>& nums, int k) {
        unordered_map<int, int> hash;
    	unordered_map<int, int>::iterator it;
    	for (size_t i = 0; i < nums.size(); i++) {
    		it = hash.find(nums[i]);
    		if (it == hash.end()) {
    			hash.insert(pair<int, int>(nums[i], i));
    		}
    		else {
    			if (i - hash[nums[i]] <= k) {
    				return true;
    			}
    			it->second = i;
    		}
    	}
    	return false;
    }
};