Leetcode 220--存在重复元素 III

最新推荐文章于 2022-06-07 10:02:54 发布

weixin_41681068

最新推荐文章于 2022-06-07 10:02:54 发布

阅读量351

点赞数

本文链接：https://blog.youkuaiyun.com/weixin_41681068/article/details/82721992

版权

本文深入探讨了滑动窗口算法在解决数组中寻找特定元素组合问题的应用，通过维护一个长度为k的窗口，实现了在时间复杂度为O(Nlogk)下判断数组中是否存在两个索引i和j，使得nums[i]和nums[j]的差的绝对值最大为t，且i和j之间的差的绝对值最大为k。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

题目：

给定一个整数数组，判断数组中是否有两个不同的索引 i 和 j，使得 nums [i] 和 nums [j] 的差的绝对值最大为 t，并且 i 和 j 之间的差的绝对值最大为 ķ。

思路：

如果使用普通的遍历方法进行遍历，则时间复杂度是O(Nk),会超出时限。我们可以换一种思路，维持一个长度为k的窗口，在这个窗口内搜索是否存在与nums[i]绝对值为t的元素，如果不存在，则更新窗口。

代码：

import java.util.Set;
import java.util.SortedSet;
import java.util.TreeSet;

public class ContainsNearbyAlmostDuplicate_2 {
	public Boolean containsNearbyAlmostDuplicate(int[] nums,int k,int t) {
		if(k<1||t<0||nums.length<=1) return false;
		
		//创建一个窗口
		SortedSet<Long> set=new TreeSet<Long>();
		for(int i=0;i<nums.length;i++) {
			//创建这个窗口的一个子集，子集的 取值范围为(nums[i]-t,nums[i]+t+1)，也就是与nums[i]相差t的元素组成的集合
			Set<Long> subSet=set.subSet((long) nums[i]-t,(long) nums[i]+t+1);
			
			//若集合不为空，则证明存在这样的元素
			if(!subSet.isEmpty()) return true;
			//下面的代码是为了维护窗口的长度为k
			if(i>=k) set.remove((long) nums[i-k]);
			set.add((long) nums[i]);
		}
		
		return false;
	}

}

创建的窗口是一个TreeSet对象，TreeSet的内部数据类型是一颗平衡树，所以对窗口的插入，删除，查找操作都是对数时间，所以代码的时间复杂度是O(Nlogk)。