532. K-diff Pairs in an Array.md

532. K-diff Pairs in an Array

思路

题目求一个数组中两个元素的值之差的绝对值为k的元素对数，首先考虑用map记录一个值是否出现，但是为了满足不重复计数，map中的值应该分成好几种情况，容易出错。
如果将数组进行排序则清晰许多，因为排序后前面的元素肯定不大于后面的元素，很容易满足不重复计数的条件。
排序后，可以用map来标记一个值是否出现，不过也可以考虑用滑动窗口的方法，这样就不用使用额外的空间而且也比map查找快。
首先考虑k不为0的情况，设置滑动窗口的左右指针left和right，初始为0和1。为了避免重复计数，将左指针右移直到不能右移(即nums[left]!=nums[left+1]), 同理右指针也应该进行同样的移动。 然后再判断nums[left] + k 和 nums[right] 的大小关系并进行合理的更新操作，具体见代码。
注意点:

• k = 0时，上述方法会失效，不过提前增加一个判断即可，见代码。
• 没说输入的k一定非负，但k应该满足非负才有意义，所以当k小于0时直接返回0即可。

C++

class Solution {
public:
    int findPairs(vector<int>& nums, int k) {
        if(k < 0) return 0;
        sort(nums.begin(), nums.end());
        int count = 0;
        if(k == 0){
            /* 这种方法较下面的方法慢一些，因为判断多一些？
            int i = 0;
            while(i < nums.size()){
                while(i < nums.size() - 1 && nums[i+1] == nums[i]) i++;
                if(i != 0 && nums[i] == nums[i-1]) count++;
                i++;
            }
            */
            int tag = 0;
            for(int i = 1; i < nums.size(); i++){
                if(nums[i] == nums[i - 1]){
                    if(tag == 0) count++;
                    tag = 1;
                }
                else tag = 0;
            }
            return count;
        }
        int left = 0, right = 1;
        while(right < nums.size()){
            while(left < nums.size() - 1 && nums[left+1] == nums[left]) left++; // 左指针右移直到不能右移
            while(right < nums.size() - 1 && nums[right+1] == nums[right]) right++; // 右指针右移直到不能右移
            if(nums[left] + k == nums[right]){
                count++;
                left++;
                right++;
            }
            else if(nums[left] + k < nums[right]) left++;
            else right++;
        }
        return count;
    }
};