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
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/contains-duplicate-ii
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
解题思路:
用哈希表来做,遍历数组nums,如果 i 不在哈希表中,就令 nums[i] 为 key,value 为当前的索引 i 。如果存在的话,判断表中的value 和 此时的 i 相差是否小于等于k,满足的话,返回True,不满足的话,更新表中的这个value。
程序代码:
class Solution:
def containsNearbyDuplicate(self, nums, k):
hashmap = {}
for i in range(len(nums)):
if nums[i] not in hashmap:
hashmap[nums[i]] = i
else:
if (i - hashmap[nums[i]] <= k ):
return True
else:
hashmap[nums[i]] = i
return False
nums = [1,2,3,1]
k = 3
s = Solution()
print(s.containsNearbyDuplicate(nums, k))