LeetCode 算法 | 数组中有重复元素吗（II）？

原创

武哥聊技术 2020-11-09 13:10:42 ©著作权

©著作权归作者所有：来自51CTO博客作者武哥聊技术的原创作品，请联系作者获取转载授权，否则将追究法律责任

精品专栏

常用经典数据结构和算法 ★★★★★
夯实Java并发基础大汇总 ★★★★★
SpringCloud微服务实战 ★★★★
2018年所有精华文章汇总 ★★★★★

阅读文本大概需要 3 分钟。

我给大家分享 LeetCode 算法题，主要都是挑选那些很可能在面试中出现的题，或者可能出现类似的题，多看看这些题至少能够在面试中不那么被动。

上次给大家分享过一次LeetCode 算法 | 数组中有重复元素吗？今天给大家分享的 LeetCode 算法题还和数组中重复元素相关，一起来看一看。

题目：

给定一个整数数组和一个整数 k，判断数组中是否存在两个不同的索引 i 和 j，使得 nums [i] = nums [j]，并且 i 和 j 的差的绝对值最大为 k。

示例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. 直接法

看到这道题，很多人的第一反应就是我挨个比较不就行了吗？首先都跟第一个元素比较，然后再第二轮，将后面的元素和第二个比较，在这过程中，只要满足题目要求，就返回 true，看下代码：

public boolean containsNearbyDuplicate(int[] nums, int k) {
  for (int i = 0; i < nums.length; i++) {
       for (int j = i + 1; j < nums.length; j++) {
           if (nums[i] == nums[j]) {
               if (j - i <= k) {
                   return true;
               }
           }
       }
   }
   return false; 
}

这种方式时间复杂度会很高，因为有两层循环，那么我们可否考虑不用两层循环呢？答案是肯定的。

2. 使用 HashMap

为什么可以使用 HashMap 呢？还记得之前我分享的一篇文章吗？LeetCode 算法 | 两数之和不简单啊，这里面有 HashMap 来处理和的问题，我在这也套用一下。

使用 num[i] 作为 key，索引的位置 i 作为 value，每次根据 num[i] 的值来获取 i，并计算间隔，如下：

public boolean containsNearbyDuplicate(int[] nums, int k) {
   Map<Integer, Integer> map = new HashMap<>();
   int inter = 0;
   for(int i = 0; i < nums.length; i++) {
       if(map.containsKey(nums[i])) {
           inter = i - map.get(nums[i]);
           if(inter <= k) {
               return true;
           }
           // 更新key为nums[i]的位置
           map.put(nums[i], i);
       } else {
           map.put(nums[i], i);
       }
   }
   return false; 
}

所以我们巧妙的运用 HashMap 可以将算法复杂度降低到 O(n)。

OK，这道算法题，就分享这么多，希望能对大家有所启发。如果你有其他思路，也欢迎留言。我建了个 LeetCode 算法交流群，大家在下方可以加我微信，我拉你进群，共同学习交流。