题目
给定一个整数数组和一个整数 k,判断数组中是否存在两个不同的索引 i 和 j,使得 nums [i] = nums [j],并且 i 和 j 的差的 绝对值 至多为 k。
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/contains-duplicate-ii
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
示例
示例 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
解答
做的时候,这个题目让我很懵逼,看了评论才明白题目是什么意思,要先找出两个相同的元素,然后再判断两个元素下标差值的绝对值是否小于等于k。
于是我就用了双指针加循环嵌套的方法来解,但是超出时间限制了。
class Solution:
def containsNearbyDuplicate(self, nums, k):
left = 0
right = 1
nums_len = len(nums)
while left <= nums_len - 2:
while right <= nums_len - 1:
if nums[left] == nums[right]:
if abs(left - right) <= k:
return True
right += 1
left += 1
right = left + 1
return False
if __name__ == '__main__':
nums = [1, 0, 1, 1]
k = 1
res = Solution().containsNearbyDuplicate(nums, k)
print(res)