开启我的LeetCode刷题日记:2006. 差的绝对值为 K 的数对数目

164 阅读2分钟

「这是我参与2022首次更文挑战的第11天,活动详情查看:2022首次更文挑战

编程世界总是离不了算法

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

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

那怎么提升呢?

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

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

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

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

今天题目:2006. 差的绝对值为 K 的数对数目

难度:简单

给你一个整数数组 nums 和一个整数 k ,请你返回数对 (i, j) 的数目,满足 i < j 且 |nums[i] - nums[j]| == k 。

|x| 的值定义为:

如果 x >= 0 ,那么值为 x 。 如果 x < 0 ,那么值为 -x 。

示例 1:

输入:nums = [1,2,2,1], k = 1 输出:4 解释:差的绝对值为 1 的数对为:

  • [1,2,2,1]
  • [1,2,2,1]
  • [1,2,2,1]
  • [1,2,2,1]

我的思路

  • 遍历nums,对于当前数num,累加上符合条件的数(num - knum + k)的数量
  • 为了快速找到符合条件的数的数量,需要用Map统计数的个数

代码实现

const countKDifference = (nums, k) => {
  let res = 0;
  const map = new Map();
  for (const num of nums) {
    res += (map.get(num - k) || 0) + (map.get(num + k) || 0);
    map.set(num, (map.get(num) || 0) + 1);
  }
  return res;
};



也可以我们进行一次遍历,遍历时下标代表 jj。对每一个 jj,我们需要知道在这个 jj 之前的符合条件的 ii 的个数,即满足 |\texttt{nums}[i] - \texttt{nums}[j]| = k∣nums[i]−nums[j]∣=k 的 ii 的个数,亦即满足 \texttt{nums}[i] = \texttt{nums}[j] + knums[i]=nums[j]+k 或 \texttt{nums}[i] = \texttt{nums}[j] - knums[i]=nums[j]−k 的 ii 的个数。使用哈希表可以在 O(1)O(1) 的时间内统计出这样的个数,因此在遍历时我们可以使用一个哈希表来维护不同数值的频率,并统计符合条件的数对总数。

总结

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

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