Leetcode-219： 存在重复元素 II【20220119打卡题】

题目

给你一个整数数组 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数组中每个元素，第二层对于遍历到的每个元素，寻找其后面的k个元素中时候存在和该元素相等的元素，有的话则返回true。

这里需要注意在第二层遍历的过程中，数组越界的问题。

C++ Code

class Solution {
public:
    bool containsNearbyDuplicate(vector<int>& nums, int k) {
        //第一层循环遍历每个元素
        //第二层循环遍历这个元素的后k个元素
        int n = nums.size();
        for(int i=0; i<n-1; i++)
        {
            int kk = i+k<n-1?i+k:n-1;
            for(int j=i+1;j<=kk;j++ )
            {
                if(nums[i]==nums[j]) return true;
            }
        }
        return false;
    }
};

然而该方法在C++中会出现超时的问题：

 思路二（哈希表）

同样是遍历数组中每个元素，如果哈希表中目前不存在该元素，则将【元素值，对应的索引】存入哈希表中。我们再哈希表中一直更新一个元素值和它对应的最大索引，如果哈希表中已存在该元素，则判断该元素的下标和当前下标的差值是否≤k，如果≤k则满足题目条件返回true，否则更新目前的这个索引（也就是当前元素对应的最大索引） ，然后继续遍历下一个元素，直到整个数组遍历完返回false。

举个例子：对于nums数组 [1,2,3,1,4,1]  k=2

• 1. 首先哈希表为空，不存在任何元素
• 2. 遍历到索引index=0，值为1的元素，将[1,0]存入哈希表当中，此时哈希表数据为 [1,0]
• 3. 遍历到索引index=1，值为2的元素，将[2,1]存入哈希表当中,此时哈希表数据为[1,0] [2,1]
• 4.遍历到索引index=2，值为3的元素，将[3,2]存入哈希表当中,此时哈希表数据为[1,0][2,1][3,2]
• 5.遍历到索引index=3，值为1的元素，哈希表中已存在key=1，下标index=0，也就是第一步中存储的数据，但此时3-0=3>k=2，不满足条件，则更新哈希表中key=1对应的值，也就是改成当前1这个元素对应的最大下标index=3，此时哈希表中数据为[1,3][2,1][3,2]
• 6.遍历到索引index=4，值为4的元素，将[4,4]存入哈希表当中,此时哈希表数据为[1,3][2,1][3,2][4,4]
• 7.遍历到索引index=5，值为1的元素，哈希表中已经存在[1,3]，此时5-3=2，成立返回true。

C++ Code

class Solution {
public:
    bool containsNearbyDuplicate(vector<int>& nums, int k) {
        unordered_map<int, int> map; // 数字，对应的最大索引
        int n = nums.size();
        for(int i=0; i<n; i++)
        {
            int num = nums[i];
            if( map.count(num) && i-map[num]<=k)  return true;//之前数组中出现过这个数字
            else map[num] = i;
        }
        return false;
    }
};