题目
Given an array of integers and an integer k, find out whether there are two distinct indices i and j in the array such that nums[i] = nums[j] and the absolute difference between i and j is at most k.
Example 1:
Input: nums = [1,2,3,1], k = 3
Output: true
Example 2:
Input: nums = [1,0,1,1], k = 1
Output: true
Example 3:
Input: nums = [1,2,3,1,2,3], k = 2
Output: false
题意
给定一个整数数组和一个整数k,找出数组中是否存在两个不同的索引i和j,使得nums [i] = nums [j],并且i和j之间的绝对差最大为k。
字典求解
刚看到这题想到的是直接用python的数组索引求解,后面发现遍历时没有相同元素是不能直接索引的。
然后使用字典的方法,将元素及索引直接存入字典中。需要注意的是,并不是元素不在字典中时才存入数据,而是所有的都存入,因为需要比较数组相同元素的下标。
python 代码
class Solution(object):
def containsNearbyDuplicate(self, nums, k):
"""
:type nums: List[int]
:type k: int
:rtype: bool
"""
if not nums:
return False
dic = dict()
for i,num in enumerate(nums):
if num in dic:
if i-dic[num] <= k:
return True
dic[num] = i
return False