20.4.10 存在重复元素Ⅱ 简单

题目

给定一个整数数组和一个整数 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. 在循环遍历过程中用map存储数字的索引，这里位置position是索引i+1，避免map初始化都是0与索引0冲突的问题；
2. 循环中，表的record如果等于0，则第一次读到这个数，把这个数的位置存进去；
3. 如果当前遍历数字的位置和已经出现过的数字的位置的差不超过k，则返回true（position - record[nums[i]] != 0，这个判断是避免相同位置的数字相减肯定等于0，但是也肯定符合不超过k的问题）；
4. 如果当前遍历数字的位置和已经出现过的数字的位置的差超过k，则更新这个数字的位置。

代码

class Solution {
public:
    bool containsNearbyDuplicate(vector<int>& nums, int k) {
        if(nums.size() == 0 || nums.size() == 1 || k == 0) return false;
        map<int, int> record;
        record[nums[0]] = 1;

        for(int i = 1, position = 2; i < nums.size(); i++, position++){
            if(record[nums[i]] == 0) record[nums[i]] = position;
            if(position - record[nums[i]] <= k && position - record[nums[i]] != 0) return true;
            else if(position - record[nums[i]] > k) record[nums[i]] = position;
        }
        return false;
    }
};