letecode [532] - K-diff Pairs in an Array 解法优-时间复杂度O(nlogn),空间O(1)

Given an array of integers and an integer k, you need to find the number of unique k-diff pairs in the array. Here a k-diff pair is defined as an integer pair (i, j), where i and j are both numbers in the array and their absolute difference is k.

Example 1:
Input: [3, 1, 4, 1, 5], k = 2
Output: 2
Explanation: There are two 2-diff pairs in the array, (1, 3) and (3, 5).
Although we have two 1s in the input, we should only return the number of unique pairs.
Example 2:
Input:[1, 2, 3, 4, 5], k = 1
Output: 4
Explanation: There are four 1-diff pairs in the array, (1, 2), (2, 3), (3, 4) and (4, 5).
Example 3:
Input: [1, 3, 1, 5, 4], k = 0
Output: 1
Explanation: There is one 0-diff pair in the array, (1, 1).
Note:
The pairs (i, j) and (j, i) count as the same pair.
The length of the array won't exceed 10,000.
All the integers in the given input belong to the range: [-1e7, 1e7].

题目大意

  给定一个整数数组和整数值k,求数组中绝对值差等于k的数对个数。

理  解:

  要求:1.求数对个数,(i,j)和(j,i) 属于同一个对,则需要注意的是(i,i)只能算一个对,尽管i出现多次

     2.首先排序数组,方便按照滑动窗口的思想解决问题

     3.设置滑动窗口的左值left=0,右值right=1:

        当nums[right]-nums[left] == k时,计数对加一;且右窗口移植至与当前right值不同的下一个位置,左窗口移植至与当前left值不同的下一个位置;

        当nums[right]-nums[left] > k时,左窗口加一left++;

        当nums[right]-nums[left] < k时,左窗口加一right++;

        若 left == right,则right右移一个。

代 码 C++:

class Solution {
public:
    int findPairs(vector<int>& nums, int k) {
        if(k<0)
            return 0;
        int count=0,left=0,right=1;
        sort(nums.begin(),nums.end());
        while(left<=right && right<nums.size()){
            if(nums[right]-nums[left] == k){
                count++;
                left++;
                right++;
                while(right<nums.size() && nums[right]==nums[right-1])
                    right++;
                while(left+1<nums.size() && nums[left]==nums[left+1])
                    left++;
                
            }else if(nums[right]-nums[left]<k){
                right++;
            }else{
                left++;
            }
            if(left==right)
                right++;
        }
        return count;
    }
};

运行结果:

  执行用时 :24 ms, 在所有 C++ 提交中击败了99.48%的用户

  内存消耗 :9.7 MB, 在所有 C++ 提交中击败了84.85%的用户
posted @ 2019-07-08 19:16  lpomeloz  阅读(192)  评论(0编辑  收藏  举报