Contains Duplicate III - LeetCode 220

题目描述:

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.

Hide Tags Binary Search Tree

分析：
此题和前一题的差别在于，需要同时判断数组元素和对应下标是否满足条件。还是借助结构体，将数组元素和对应的下标存放到一个结构体数组中，然后对结构体数组进行一级排序，之后利用双指针判断是否存在值元素相差不大于t且对应下标相差不大于k。
实现时，在判断值相差是否不大于t的时候，要注意防止整型溢出，所以应该将其变形为等价的表达式，即判断较小元素与t的和是否不小于于较大元素。

以下是C++实现代码：

*///20ms///*/
struct Node{
    int num; //数组元素
    int index;  //数组元素对应下标
};
int cmp(Node &a,Node &b) //结构体一级排序函数
{
    return a.num < b.num;
}

class Solution {
public:
    bool containsNearbyDuplicate(vector<int>& nums, int k) {
        int size = nums.size();
        if(k <= 0)
            return false;
        vector<Node> vec;
        for(int i = 0; i != size; i++) //将nums中的元素放入一个结构体数组中
        {
            Node nod;
            nod.num = nums[i];
            nod.index = i;
            vec.push_back(nod);
        }
        sort(vec.begin(),vec.end(),cmp); //非递减排序
         
        int i = 0, j = 1; //以下是双指针查找，i，j分别为左右指针
	while( i < j && j != size )
        {			
		if(vec[j].num <= t + vec[i].num) //防止整型溢出
		{
			if(abs(vec[j].index - vec[i].index)<=k) //找到满足条件的数
				return true;
				j++;
			if(i < j && j == size) //右指针已超出数组下标，表示数组的num是满足条件的，但是下标不满足，所以修改左右指针，继续下一个循环
			{ 
			    i++;
			    j = i+1;
		        }
		}
		else //数组元素就不满足，直接将两个指针同时右移一个位置
		{
	            i++;
		    j = i+1;
		}        			
	}        
        return false;
    }
};