220. 存在重复元素 III
在整数数组 nums 中,是否存在两个下标 i 和 j,使得 nums [i] 和 nums [j] 的差的绝对值小于等于 t ,且满足 i 和 j 的差的绝对值也小于等于 ķ 。如果存在则返回 true,不存在返回 false。
示例 1:
输入: nums = [1,2,3,1], k = 3, t = 0
输出: true
示例 2:
输入: nums = [1,0,1,1], k = 1, t = 2
输出: true
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/contains-duplicate-iii
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/contains-duplicate-iii
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
桶排序+哈希表
这题真难,上当了。。。
此题与【219. 存在重复元素 II】的区别在于:
1.nums [i] 和 nums [j] 的差的绝对值小于等于 t。
2.i和j可以相同
本来我试图修改【219. 存在重复元素 II】的代码做这道题,但是因为求的是差值,对于每一个数来说其他数和它的差值都是意义不同的,所以一个字典不够,每个数都需要一个字典(参照【447. 回旋镖的数量】,时间复杂度O(N^2))。暴力算法可以修改成功,都是依旧超时,显然此题还是想要O(n)的时间复杂度。
class Solution:
def containsNearbyAlmostDuplicate(self, nums: List[int], k: int, t: int) -> bool:
if t<0 or not nums:
return False
for i in range(max(1,len(nums)-k)):
tmp = sorted(nums[i:i+k+1])
for j in range(len(tmp)-1):
if tmp[j+1]-tmp[j]<=t:
return True
return False
那么单单哈希算法是不够的,需要桶排序进一步减少搜索范围。
class Solution:
def containsNearbyAlmostDuplicate(self, nums: List[int], k: int, t: int) -> bool:
dic={}
if t<0 or not nums :
return False
for i,num in enumerate(nums):
idx=num//(t+1)
if idx in dic:
return True
if idx-1 in dic and abs(num-dic[idx-1])<=t:
return True
if idx+1 in dic and abs(num-dic[idx+1])<=t:
return True
dic[idx]=num
if i>=k:
dic.pop(nums[i-k]//(t+1))
return False