[LeetCode练习]219. 存在重复元素 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

思路:1、找到相同的值,比较数组的索引,只要满足|i-j|<=k即可 

代码

public static boolean containsNearbyDuplicate(int[] nums, int k) {
    Map<Integer, String> map = new HashMap<>();
    if (nums == null) {
        return false;
    }
    //将相同的值得索引统一存放
    for (int i = 0; i < nums.length; i++) {
        if (map.containsKey(nums[i])) {
            String str = map.get(nums[i]) + "," + i;
            map.put(nums[i], str);
        } else {
            map.put(nums[i], String.valueOf(i));
        }
    }
    //对每个value找存在差为k的
    Iterator iterator = map.values().iterator();
    while (iterator.hasNext()) {
        if (isK((String) iterator.next(), k)) {
            return true;
        }
    }
    return false;
}

private static boolean isK(String str, int k) {
    String[] strs = str.split(",");
    int i = 0, j = 1;
    if (strs.length == 1) {
        return false;
    }
    while (i<strs.length-1) {
        if (Math.abs(Integer.valueOf(strs[i]) - Integer.valueOf(strs[i+1])) <= k) {
            return true;
        }
        i++;
    }
    return false;
}

猜你喜欢

转载自blog.csdn.net/lbh199466/article/details/86562259