LeetCode 219——存在重复的元素II

一、题目介绍

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

二、解题思路

本题可以使用哈希映射map结果。遍历数组，如果当前的元素在map中找不到，则将其插入并记录其在数组中的下标位置，如果能找到分为以下两种情况：

（1）当前元素所处下标位置 i - map[nums[i]] <= k，则返回true。

（2）当前元素所处下标位置 i - map[nums[i]] > k，则更新map[nums[i]]的值，然后继续遍历。

三、解题代码

class Solution {
public:
    bool containsNearbyDuplicate(vector<int>& nums, int k) {
        if(k == 0) //要求两个不同的索引
            return false;
        else
        {
            map<int, int> mp;
            for(int i = 0; i < nums.size(); ++i)
            {
                if(mp.count(nums[i]) == 1) //注意下标为0的情况
                {
                    if(i - mp[nums[i]] <= k)
                        return true;
                    else //如果 i 和 j 之间的差值不等于k，则将当前元素对应的索引更新
                        mp[nums[i]] = i;
                }
                else
                {
                    mp[nums[i]] = i;
                }
            }
            return false;
        }
        
    }
};

四、解题结果