存在重复元素 II

题目描述

给定一个整数数组和一个整数 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

分析

定义HashMap，其中键是数组中的数字，值是数字的索引组成的集合。
取得值的索引之差的最小值，与k比较

代码

class Solution {
    public boolean containsNearbyDuplicate(int[] nums, int k) {
		HashMap<Integer,List<Integer>> map = new HashMap<>();
		for(int i=0;i<nums.length;i++) {
			List<Integer> list = new ArrayList<>();
			list.add(i);
			if(map.containsKey(nums[i])) {
				List<Integer> list2 = new ArrayList<>();
				list2.addAll(map.get(nums[i]));
				list2.add(i);
				map.put(nums[i], list2);
				continue;
			}
			map.put(nums[i], list);
		}
        int min=Integer.MAX_VALUE;
		for(int j:map.keySet()){
            List<Integer> list3 = map.get(j);
            int size=list3.size();
            if(size>1){
                for(int a=0;a<size-1;a++){
                    min=Math.min(list3.get(a+1)-list3.get(a),min);
                }
            }
        }
        return min<=k;
    }
}

优化

leetcode评论区大神的方法

public boolean containsNearbyDuplicate(int[] nums, int k) {
        HashMap<Integer, Integer> map = new HashMap<>();
        for (int i = 0; i < nums.length; i++) {
            Integer index = map.get(nums[i]);
            if (index == null) {
                map.put(nums[i], i);
            } else if (i - index <= k) {
                return true;
            } else {
                map.put(nums[i], i);
            }
        }
        return false;
    }