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

这个题和217题类似,只是多了一个条件,所以这里用map哈希表存下元素及其索引,然后找到相同元素时判断是否与当前索引相差值小于k,是则返回true,如果没有相同元素则继续添加到表中。
C++源代码:

class Solution {
public:
    bool containsNearbyDuplicate(vector<int>& nums, int k) {
        map<int, int> m;
        for(int i=0;i<nums.size();i++)
        {
            if(m.find(nums[i])!=m.end() && i-m[nums[i]]<=k)
                return true;
            else
                m[nums[i]] = i;
        }
        return false;
    }
};

python3源代码:

class Solution:
    def containsNearbyDuplicate(self, nums, k):
        """
        :type nums: List[int]
        :type k: int
        :rtype: bool
        """
        m = {}
        for i in range(len(nums)):
            if nums[i] in m and i-m[nums[i]]<=k:
                return True
            else:
                m[nums[i]] = i
        return False

猜你喜欢

转载自blog.csdn.net/x603560617/article/details/84024653
今日推荐