219存在重複元素 II(HashMap、哈希表+滑動窗口)

1、題目描述

給定一個整數數組和一個整數 k,判斷數組中是否存在兩個不同的索引 i 和 j,使得 nums [i] = nums [j],並且 i 和 j 的差的 絕對值 至多爲 k。

2、示例 

輸入: nums = [1,2,3,1], k = 3
輸出: true

輸入: nums = [1,0,1,1], k = 1
輸出: true

3、題解

解法一:

基本思想:HashMap,map保存nums中元素對應下標位置的映射。遍歷nums所有元素,對於當前元素nums[i],如果存在於HashMap,計算當前下標i與HashMap[nums[i]]的距離,如果小於等於k返回true,同時更新HashMap[nums[i]]爲當前下標i。

解法二:

基本思想:哈希表+滑動窗口,用哈希表來維護這個k大小的滑動窗口,如果下一個元素存在於滑動窗口內,返回true,否則加入到滑動窗口,同時滑動窗口去掉最舊的元素。

#include<iostream>
#include<vector>
#include<algorithm>
#include<unordered_map>
#include<unordered_set>
using namespace std;
class Solution {
public:
	bool containsNearbyDuplicate(vector<int>& nums, int k) {
		//基本思想:HashMap,map保存nums中元素對應下標位置的映射
		//遍歷nums所有元素,對於當前元素nums[i],如果存在於HashMap,計算當前下標i與HashMap[nums[i]]的距離
		//如果小於等於k返回true,同時更新HashMap[nums[i]]爲當前下標i
		unordered_map<int, int> HashMap;
		for (int i = 0; i < nums.size(); i++)
		{
			if (HashMap.find(nums[i]) != HashMap.end())
			{
				if (i - HashMap[nums[i]] <= k)
					return true;
			}
			HashMap[nums[i]] = i;
		}
		return false;
	}
};
class Solution1 {
public:
	bool containsNearbyDuplicate(vector<int>& nums, int k) {
		//基本思想:哈希表+滑動窗口,用哈希表來維護這個k大小的滑動窗口
		//如果下一個元素存在於滑動窗口內,返回true,否則加入到滑動窗口,同時滑動窗口去掉最舊的元素
		unordered_set<int> container;
		for (int i = 0; i < nums.size(); i++)
		{
			if (container.find(nums[i]) != container.end())
				return true;
			container.insert(nums[i]);
			if (container.size() > k)
				container.erase(nums[i - k]);
		}
		return false;
	}
};
int main()
{
	Solution1 solute;
	vector<int> nums = { 1,2,3,1,2,3 };
	int k = 2;
	cout << solute.containsNearbyDuplicate(nums, k) << endl;
	return 0;
}

 

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章