LeetCode 220. 存在重复元素 III--Set简单处理

在整数数组 nums 中，是否存在两个下标 i 和 j，使得 nums [i] 和 nums [j] 的差的绝对值小于等于 t ，且满足 i 和 j 的差的绝对值也小于等于 ķ 。

如果存在则返回 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

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/contains-duplicate-iii
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

题解：从头开始处理，把元素存入set容器中，保证容器中的元素个数小于等于k，然后每次去找容器中是否满足和当前的nums[i]绝对值差小于等于t的元素，也就是在容器中找和nums[i]最近的元素，需要同时返回大于等于nums[i]和小于等于nums[i]的元素，然后把nums[i]插入容器，如果容器元素大于k,则在容器中找到nums[i-k]并删除。

AC代码

class Solution {
public:
	typedef long long ll;
	bool containsNearbyAlmostDuplicate(vector<int>& nums, int k, int t) 
	{
		if(k==0)return false;
		set<int>q;
		set<int>::iterator it;
		q.insert(nums[0]);
		for(int i=1;i<nums.size();i++)
		{
			it=--q.upper_bound(nums[i]);
			if(it!=q.end())
			{
				if(abs((ll)*it-(ll)nums[i])<=(ll)t)return true;
			}
			it=q.lower_bound(nums[i]);
			if(it!=q.end())
			{
				if(abs((ll)*it-(ll)nums[i])<=(ll)t)return true;
			}
			q.insert(nums[i]);
			if(q.size()>k)
			{
				it=q.lower_bound(nums[i-k]);
				q.erase(it);
			}
		}
		return false;
	}
};