LeetCode——存在重复元素(二)
题目描述:
给定一个整数数组和一个整数 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
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/contains-duplicate-ii
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
解题思路:
我们定义一个dic字典,用来存储nums中出现过的数字及其下标。
我们遍历整个nums,如果dic[ nums[ i ] ]不存在,我们就将其及其下标 i 添加进dic中。如果已经存在,我们就判断两个的下标差的绝对值是否小于等于k,如果满足则返回True,如果大于k我们就更新dic中nums[ i ]的下标。
python代码:
class Solution(object):
def containsNearbyDuplicate(self, nums, k):
"""
:type nums: List[int]
:type k: int
:rtype: bool
"""
dic = {}
for i in range(len(nums)):
if nums[i] not in dic:
dic[nums[i]] = i
elif abs(i - dic[nums[i]]) <= k:
return True
else:
dic[nums[i]] = i
return False