1. 程式人生 > >Leetcode 220.存在重複元素III

Leetcode 220.存在重複元素III

存在重複元素III

給定一個整數陣列,判斷陣列中是否有兩個不同的索引 ij,使得 nums [i] 和 nums [j] 的差的絕對值最大為 t,並且 ij 之間的差的絕對值最大為 ķ

示例 1:

輸入: nums = [1,2,3,1], k= 3, t = 0

輸出: true

示例 2:

輸入: nums = [1,0,1,1], k=1, t = 2

輸出: true

示例 3:

輸入: nums = [1,5,9,1,5,9], k = 2, t = 3

輸出: false

 1 import
java.util.TreeSet; 2 3 public class Solution{ 4 public boolean containsNearbyAlmostDuplicate(int[] nums,int k,int t){ 5 if(nums==null||nums.length==0||k<=0) return false; 6 TreeSet<Long> ts=new TreeSet(); 7 for(int i=0;i<nums.length;i++){ 8 Long right=ts.floor((long
)nums[i]+t); 9 Long left=ts.ceiling((long)nums[i]-t); 10 if(right!=null&&left!=null&&right>=left) return true; 11 ts.add((long)nums[i]); 12 if(i>=k)//i>k說明nums[i]之前至少已經有k個不相同的元素(否則早就返回true) 13 ts.remove((long)nums[i-k]);
14 } 15 return false; 16 } 17 }