Contains Duplicate III - LeetCode 220

来源:互联网 发布:2016年淘宝双十一销售额 编辑:程序博客网 时间:2024/05/18 02:30
题目描述:

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;    }};


0 0