leetcode(532)-- K-diff Pairs in an Array

最新推荐文章于 2021-02-28 12:25:15 发布

linjiet

最新推荐文章于 2021-02-28 12:25:15 发布

阅读量102

点赞数

分类专栏：算法文章标签： leetcode 532

本文链接：https://blog.csdn.net/qq_39742013/article/details/84074764

版权

算法专栏收录该内容

33 篇文章 0 订阅

订阅专栏

class Solution {
public:
    int findPairs(vector<int>& nums, int k) {
        if(k<0)
            return 0;
        //考虑k为0，不能去重
        int result=0;
        sort(nums.begin(),nums.end());
        int size=nums.size();
        /*
        //先去重
        int curInsert=1;
        for(int i=1;i<size;i++)
        {
            if(nums[i]==nums[i-1])
                continue;
            else
            {
                nums[curInsert]=nums[i];
                curInsert++;
            }
        }
        */
        /*
        //解1
        for(int i=0;i<size;i++)
        {
            if(i!=0&&nums[i-1]==nums[i])
                continue;
            int expect=nums[i]+k;
            for(int j=i+1;j<size;j++)
            {
                if(nums[j]==expect)
                {
                    result++;break;
                }
                else if(nums[j]>expect)
                    break;
            }
            
        }
        return result;
        */
        //解2
        int j=1;
        for(int i=0;i<size;i++)
        {
            if(i!=0&&nums[i-1]==nums[i])
                continue;
            int expect=nums[i]+k;
            for(;j<size;j++)
            {
                //我们需要j>i,但是如例子[1,1,3,4,5]，k=0，这时候当i=2时，j也等于2
                if(i==j)j=i+1; 
                if(nums[j]==expect)
                {
                    result++;
                    while(nums[j]==expect&&j<size)j++;
                    break;
                }
                else if(nums[j]>expect)
                    break;
            }
            
        }
        return result;
    }
};

后面这种解法不是对任何输入样例都有很好的解，对k比较大且nums中的数字差别很小时的情况效果较好。而如果k=0这种情况下，那么效果不及解法1。

总结：

1.要把题目中描述的变量的取值范围弄清楚；

2.要把题目的要求进行总结然后得出一般规律。