[刷题] 220 Contains Duplicate III

要求

• 给出整型数组nums和整数k，是否存在索引i和j
• 使得nums[i]和nums[j]之差不超过t，且i和j之差不超过k

思路

• 建立k个元素的有序查找表
• 每次有新元素加入，寻找查找表中大于 nums[i]-t 的最小值，若存在且此值小于 nums[i]+t，则目标元素存在
• set底层是二叉树实现，时间（nlogn），空间（k）
• vector中的int值相加可能产生整型溢出，set中使用64位整数 long long

实现

class Solution {
public:
    bool containsNearbyAlmostDuplicate(vector<int>& nums, int k, int t) {
        if(t < 0)
            return false;

        set<long long> record;
        for(int i = 0 ; i < nums.size() ; i ++){

            if(record.lower_bound((long long)nums[i] - (long long)t) != record.end() &&
                *record.lower_bound((long long)nums[i] - (long long)t ) <= (long long)nums[i] + (long long)t)
                return true;

            record.insert(nums[i]);

            if(record.size() == k + 1)
                record.erase( nums[i-k] );
        }

        return false;
    }
};