JavaScript——leetcode2006 差的绝对值为k的数对数目

JavaScript——leetcode2006差的绝对值为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]

示例 2：
输入：nums = [1,3], k = 3
输出：0
解释：没有任何数对差的绝对值为 3 。

示例 3：
输入：nums = [3,2,1,5,4], k = 2
输出：3
解释：差的绝对值为 2 的数对为：

• [3,2,1,5,4]
• [3,2,1,5,4]
• [3,2,1,5,4]

代码片

方法一：暴力方法 两次循环

var countKDifference = function(nums, k) {
    let res = 0, n = nums.length;
    for(let i =0; i<n; ++i){
        for(let j = i+1; j<n; ++j){
            if(Math.abs(nums[i] - nums[j]) == k){
                res++;
            }
        }
    }
    return res;
};

方法二：哈希表+一次遍历

var countKDifference = function(nums, k) {
    let res = 0,n=nums.length;
    const cnt = new Map();
    //j代表遍历时的下标
    for(let j = 0; j<n; j++){
        res += (cnt.get(nums[j] - k) || 0) + (cnt.get(nums[j] + k) || 0);
        cnt.set(nums[j],(cnt.get(nums[j]) || 0) +1);
    }
    return res;
};