LeetCode - 532 - 数组中的K-diff数对（k-diff-pairs-in-an-array）

最新推荐文章于 2024-05-31 10:21:05 发布

JavaScriptLiang

最新推荐文章于 2024-05-31 10:21:05 发布

阅读量147

点赞数

本文链接：https://blog.csdn.net/weixin_41806099/article/details/102982572

版权

这篇博客主要讲解了LeetCode中的第532题——K-diff数对。文章介绍了问题的难度、涉及的知识点，并提供了解题思路和代码实现。作者首先分享了暴力求解的方法，然后通过优化提高了解决方案的效率。最后，鼓励读者参与讨论并持续学习前端与算法知识。

摘要由CSDN通过智能技术生成

一目录

不折腾的前端，和咸鱼有什么区别

目录
一目录
二前言
三解题及测试
四 LeetCode Submit
五解题思路
六进一步思考

二前言

难度：简单
涉及知识：数组、双指针
题目地址：https://leetcode-cn.com/problems/k-diff-pairs-in-an-array/
题目内容：

给定一个整数数组和一个整数 k, 你需要在数组里找到不同的 k-diff 数对。

这里将 k-diff 数对定义为一个整数对 (i, j),

其中 i 和 j 都是数组中的数字，且两数之差的绝对值是 k.

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

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

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

注意:
数对 (i, j) 和数对 (j, i) 被算作同一数对。
数组的长度不超过10,000。
所有输入的整数的范围在 [-1e7, 1e7]。

三解题及测试

小伙伴可以先自己在本地尝试解题，再回来看看 jsliang 的解题思路。

LeetCode 给定函数体：

/**
 * @param {number[]} nums
 * @param {number} k
 * @return {number}
 */
var findPairs = function(nums, k) {
    
};

根据上面的已知函数，尝试破解本题吧~

确定了自己的答案再看下面代码哈~

index.js

/**
 * @name 数组中的K-diff数对
 * @param {number[]} nums
 * @param {number} k
 * @return {number}
 */
const findPairs = (nums, k) => {
  const map = new Map();
  let result = 0;
  for (let i = 0; i < nums.length - 1; i++) {
    for (let j = i + 1; j < nums.length; j++) {
      const max = Math.max(nums[i], nums[j]);
      const min = Math.min(nums[i], nums[j]);
      if (max - min === k && map.get(`${min}${max}`) === undefined) {
        result += 1;
        map.set(`${min}${max}`, `${min}${max}`,);
      }
    }
  }
  return result;
};

// console.log(findPairs([3, 1, 4, 1, 5], 2)); // 2
// console.log(findPairs([1, 2, 3, 4, 5], 1)); // 4
console.log(findPairs([1, 3, 1, 5, 4], 0)); // 1

node index.js 返回：

四 LeetCode Submit

Accepted
* 72/72 cases passed (2584 ms)
* Your runtime beats 5.22 % of javascript submissions
* Your memory usage beats 6.25 % of javascript submissions (40 MB)

五解题思路

首先，拿到题目，本想秀一秀，使用双重 Map，但是霎时间失败，只好先使用暴力解题先：

const findPairs = (nums, k) => {
  const map = new Map();
  let result = 0;
  for (let i = 0; i < nums.length - 1; i++) {
    for (let j = i + 1; j < nums.length; j++) {
      const max = Math.max(nums[i], nums[j]);
      const min = Math.min(nums[i], nums[j]);
      if (max - min === k && map.get(`${min}${max}`) === undefined) {
        result += 1;
        map.set(`${min}${max}`, `${min}${max}`,);
      }
    }
  }
  return result;
};

Submit 结果 “非常不错”：

Accepted
* 72/72 cases passed (2584 ms)
* Your runtime beats 5.22 % of javascript submissions
* Your memory usage beats 6.25 % of javascript submissions (40 MB)

然后，你也知道，俺是喜欢较真的人，双重遍历实在太垃圾了，咱们试试优化一下：

/**
 * @name 数组中的K-diff数对
 * @param {number[]} nums
 * @param {number} k
 * @return {number}
 */
const findPairs = (nums, k) => {
  if (k < 0) {
    return 0;
  }
  const map1 = new Map();
  const map2 = new Map();
  let result = 0;
  for (let i = 0; i < nums.length; i++) {
    if (nums[i] + k === map1.get(nums[i] + k) && map2.get(`${nums[i]}${nums[i] + k}`) === undefined) {
      result += 1;
      map2.set(`${nums[i]}${nums[i] + k}`, `${nums[i]}${nums[i] + k}`);
    }
    if (nums[i] - k === map1.get(nums[i] - k) && map2.get(`${nums[i] - k}${nums[i]}`) === undefined) {
      result += 1;
      map2.set(`${nums[i] - k}${nums[i]}`, `${nums[i] - k}${nums[i]}`);
    }
    map1.set(nums[i], nums[i]);
  }
  return result;
};

Submit 试试：

Accepted
* 72/72 cases passed (92 ms)
* Your runtime beats 83.48 % of javascript submissions
* Your memory usage beats 6.25 % of javascript submissions (41.7 MB)

最后，有更好 idea 的小伙伴，欢迎评论吐槽或者直接私聊我咯~

不折腾的前端，和咸鱼有什么区别！

640?wx_fmt=png