220. Contains Duplicate III / set高阶用法

题目描述

Given an array of integers, find out whether there are two distinct indices i and j in the array such that the absolute difference between nums[i] and nums[j] is at most t and the absolute difference between i and j is at most k.

Example 1:

Input: nums = [1,2,3,1], k = 3, t = 0
Output: true

Example 2:

Input: nums = [1,0,1,1], k = 1, t = 2
Output: true

Example 3:

Input: nums = [1,5,9,1,5,9], k = 2, t = 3
Output: false

解析

class Solution {
public:
    bool containsNearbyAlmostDuplicate(vector<int>& nums, int k, int t) {
        set<long long> hash;
        for(int i=0;i<nums.size();i++){
            auto it=hash.lower_bound((long long)nums[i]-(long long)t);
            if(it!=hash.end()&&*it<=(long long)nums[i]+(long long)t) return true;
            hash.insert(nums[i]);
            if(hash.size()>k) hash.erase(nums[i-k]);
        }
        return false;
    }
};
发布了123 篇原创文章 · 获赞 11 · 访问量 5554

猜你喜欢

转载自blog.csdn.net/weixin_43590232/article/details/104525078