LeetCode219.存在重复元素2Golang版

LeetCode219.存在重复元素2Golang版

1. 问题描述

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

2. 思路

同样初始化一个map，这里并不存放数组元素出现的次数，而是存放数组元素的索引值。如果在元素nums[i]之前k个位置以内，出现过该元素，则返回true。

注意由于初始化的map中，默认值为0。如果map中存放的索引仍然从0开始，会发生混淆，索引map中存放索引值时+1，遍可以通过判断map中元素是否为0来判断之前是否出现过某元素。

3. 代码

func containsNearbyDuplicate(nums []int, k int) bool {
    // 不是直接存放出现的次数，而是存放索引值+1
    charAt := map[int]int{}
    for i := 0; i < len(nums); i++ {
        if charAt[nums[i]] != 0 && i - (charAt[nums[i]] - 1) <= k {
            return true
        }
        charAt[nums[i]] = i + 1
    }
    return false
}