220. 存在重复元素 III

 

思路

难度中等645收藏分享切换为英文接收动态反馈

给你一个整数数组 nums 和两个整数 k 和 t 。请你判断是否存在 两个不同下标 i 和 j，使得 abs(nums[i] - nums[j]) <= t ，同时又满足 abs(i - j) <= k 。

如果存在则返回 true，不存在返回 false。

 

示例 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

 

 

 

from sortedcontainers import SortedList

class Solution:
    def containsNearbyAlmostDuplicate(self, nums: List[int], k: int, t: int) -> bool:
        # O(N logk)
        def find(nums,target):
            lo = 0
            n = len(nums)
            hi = n -1
            while lo <=hi:
                mid = lo+(hi-lo)//2
                if nums[mid] == target:
                    return mid
                elif nums[mid] > target:
                    hi = mid -1
                else:
                    lo = mid +1
            return lo
        window = SortedList()
        #是基于红黑树实现的，快速的插入和删除操作，时间复杂度为 O(log n)
        for i in range(len(nums)):
            if i > k:
                window.remove(nums[i - 1 - k])
            window.add(nums[i])
            idx = find(window, nums[i])
            #那么返回的索引将是第一个与 nums[i] 相等的元素的位置。如果 nums[i] 不存在于列表中，返回的索引将是将 nums[i] 插入到列表中的位置。

            # 检查 idx+1 与idx   idx 与idx -1 
            if idx > 0 and abs(window[idx] - window[idx-1]) <= t:
                return True
            if idx < len(window) - 1 and abs(window[idx+1] - window[idx]) <= t:
                return True
        return False

 

 

 

 

 

 

 

 

 

暴力：

class Solution {
public:
    bool containsNearbyAlmostDuplicate(vector<int>& nums, int k, int t) {
        
        for(int i = 0; i < nums.size();i++) {            
            for (int j = i+1;j <= i+k;j++) {
                if (j < nums.size() && abs(long(nums[i])-long(nums[j])) <=t ) {
                return true;
                }  
            }
        }
        return false;
    }
};

 

 

 

 

 

 

 

 

 

class Solution {
public:
    bool containsNearbyAlmostDuplicate(vector<int>& nums, int k, int t) {
        set<long> st; 
        for(int i = 0; i < nums.size();i++) {
            auto lb = st.lower_bound(long(nums[i])-t);
            if (lb != st.end() && *lb <= (long)nums[i] + t) return true;
            st.insert(nums[i]);
            if (i-k>=0) {
                st.erase(nums[i-k]);
            }            
        }
        return false;
    }
};