leetcode - Contains Duplicate III

题目：

Contains Duplicate III

 

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

分析：

    利用multiset进行BST的二分搜索。lower_bound()返回一个迭代器，指向大于等于输入值的第一个元素。

    从左到右扫描数组，若multiset的元素数量等于k+1，则删去最先进入的元素。由于multiset.size()始终小于等于k+1，所以下标之差是肯定小于等于k的。然后利用lower_bound()，找到第一个大于等于nums[i]-t的元素，若该元素和nums[i]的差的绝对值小于等于t，则返回真。

  注意：为了防止溢出，必须将数据转化为long long进行处理！

class Solution {public:    bool containsNearbyAlmostDuplicate(vector<int>& nums, int k, int t) {        if(nums.size()<=1 || k<1 || t<0)            return false;                    multiset<long long> mset;        long long long_t=t;                for(int i=0;i<nums.size();++i)        {            if(mset.size()==k+1)            {                auto it=mset.find(nums[i-k-1]);                mset.erase(it);            }            long long tmp=nums[i];            auto it=mset.lower_bound(tmp-long_t);            if(it!=mset.end())            {                long long diff=*it>tmp?*it-tmp:tmp-*it;                if(diff<=long_t)                    return true;            }            mset.emplace(nums[i]);        }        return false;    }};