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

本文介绍了一种高效算法,用于解决在给定整数数组中查找特定差值k的唯一数对数量的问题。通过排序和滑动窗口策略,算法能够在O(n log n)时间内找到所有满足条件的数对。

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%的用户

转载于:https://www.cnblogs.com/lpomeloz/p/11153174.html

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值