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;
}