LeetCode刷题日记:532. 数组中的 k-diff 数对

63 阅读1分钟

持续创作,加速成长!这是我参与「掘金日新计划 · 6 月更文挑战」的第29天,点击查看活动详情

编程世界总是离不了算法

最近在看框架源码时,会有很多算法的实现逻辑,有时候会感到吃力

于是决定蹭着假期,加强算法和数据结构相关的知识

那怎么提升呢?

其实我知道算法这东西没有捷径,多写多练才能提升,于是我开启我的LeetCode刷题之旅

第一阶段目标是:200道,每天12

为了不乱,本系列文章目录分为三部分:

  1. 今日题目:xxx
  2. 我的思路
  3. 代码实现

532. 数组中的 k-diff 数对

给你一个整数数组 nums 和一个整数 k,请你在数组中找出 不同的 k-diff 数对,并返回不同的 k-diff 数对 的数目。

k-diff 数对定义为一个整数对 (nums[i], nums[j]) ,并满足下述全部条件:

0 <= i, j < nums.length i != j nums[i] - nums[j] == k 注意,|val| 表示 val 的绝对值。

 

示例 1:

输入:nums = [3, 1, 4, 1, 5], k = 2
输出:2
解释:数组中有两个 2-diff 数对, (1, 3) 和 (3, 5)。
尽管数组中有两个 1 ,但我们只应返回不同的数对的数量。

示例 2:

输入:nums = [1, 2, 3, 4, 5], k = 1
输出:4
解释:数组中有四个 1-diff 数对, (1, 2), (2, 3), (3, 4) 和 (4, 5) 。

示例 3:

输入:nums = [1, 3, 1, 5, 4], k = 0
输出:1
解释:数组中只有一个 0-diff 数对,(1, 1) 。

提示:

1 <= nums.length <= 104
-107 <= nums[i] <= 107
0 <= k <= 107

来源:力扣(LeetCode) 链接:leetcode.cn/problems/k-… 著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。

我的思路

1.本题解使用了二分查找的方法。

2.分析题目,本题要求找出nums[i] - nums[j] === k的不同数对的数目。分析这个公式,我们可以将其变形nums[i] - k === nums[j],我们可以通过遍历nums对每一个下标i找满足公式的nums[j]。这样我们就把题目转化成了找target的题目,这种时候我们很容易联想到二分查找,并且由于是找数对,所以原数组的顺序并不重要,所以我们可以将nums排序,因此我们可以使用二分查找。

3.进行二分查找时,由于对于每一次遍历的下标i,它实际上是数对的右边界,因此应该将i作为二分查找的右边界。并且注意while的条件(这里可以记为一个模版,当右边界取得到的时候就使用left < right,也可以通过分析得出,因此如果取left <= right,则二分查找会陷入死循环)。

4.最后要小心题目的找不同的数对(数值不同),因此我们可以使用一个哈希集合储存找到的满足条件的nums[j](因为nums[j]一确定根据等式,那么nums[i]也就确定了,所以不能有相同的nums[j])。

代码实现

/**
 * @param {number[]} nums
 * @param {number} k
 * @return {number}
 */
var findPairs = function(nums, k) {
    nums.sort((a,b) => a-b);
    let set = new Set();
    for(let i = 0;i < nums.length;i++)
    {
        const j = binarySearch(nums,i,nums[i] - k);
        if(j !== null)
        {
            set.add(nums[j])
        }
    }
    return set.size;
};

function binarySearch(nums,end,target)
{
    let left = 0;
    let right = end;
    while(left < right)
    {
        let mid = Math.floor((left + right) / 2);
        if(nums[mid] > target)
        {
            right = mid;
        }
        else if(nums[mid] < target){
            left = mid + 1;
        }
        else{
            return mid;
        }
    }
    return null;
}

总结

实现方式其实有很多,这里仅供参考~

由于刚开始刷题,也不知道从哪里刷好,如果前辈们有好的建议,希望不吝赐教,感谢🌹