LeetCode 2006. 差的绝对值为 K 的数对数目题解

最新推荐文章于 2024-08-23 11:46:54 发布

小明不要写Bug

最新推荐文章于 2024-08-23 11:46:54 发布

阅读量212

点赞数

分类专栏： LeetCode 每日一题题解文章标签： leetcode 算法 c++ java 哈希

本文链接：https://blog.csdn.net/qq_46176960/article/details/122843733

版权

LeetCode 每日一题题解专栏收录该内容

92 篇文章 2 订阅

订阅专栏

2006. 差的绝对值为 K 的数对数目题解

题目来源：2006. 差的绝对值为 K 的数对数目

2022.02.09 每日一题

要在数组 nums 中寻找 i < j 的同时满足 | nums[ i ] - nums [ j ] | == k 的数对的数目

法一：暴力模拟

循环两遍数组，寻找满足条件的数对

class Solution {
public:
    int countKDifference(vector<int> &nums, int k) {
        // 统计 nums 的长度
        int len = nums.size();
        // 如果长度为 1 则不存在 i < j 可以直接返回 0
        if (len == 1) return 0;
        // 将结果初始化为 0
        int res = 0;
        // 遍历两遍数组 nums[i] - nums[j] 的绝对值 与 k 相等
        for (int i = 0; i < len - 1; i++) {
            for (int j = i + 1; j < len; j++) {
                // 如果满足 nums[i] - nums[j] 的绝对值 与 k 相等
                // 结果 res 加一
                if (abs(nums[i] - nums[j]) == k) {
                    res++;
                }
            }
        }
        return res;
    }
};

class Solution {
    public int countKDifference(int[] nums, int k) {
        // 统计 nums 的长度
        int len = nums.length;
        // 如果长度为 1 则不存在 i < j 可以直接返回 0
        if (len == 1) return 0;
        // 将结果初始化为 0
        int res = 0;
        // 遍历两遍数组 nums[i] - nums[j] 的绝对值 与 k 相等
        for (int i = 0; i < len - 1; i++) {
            for (int j = i + 1; j < len; j++) {
                // 如果满足 nums[i] - nums[j] 的绝对值 与 k 相等
                // 结果 res 加一
                if (Math.abs(nums[i] - nums[j]) == k) {
                    res++;
                }
            }
        }
        return res;
    }
}

时间复杂度O( $n^2$ )
空间复杂度O(1)

法二：一次遍历+哈希表

创建一个哈希表，统计每个数字出现的次数，

由于 $∣ n u m s [i] - n u m s [j] ∣ = k$ 可以拆解出$ nums[i] - nums[j] = k $与$ nums[i] - nums[j] = -k $

因此可以转换为 $n u m s [j] = n u m s [i] + k$ 和 $n u m s [j] = n u m s [i] - k$

所以就可以通过查询 $n u m s [i] + k$ 与 $n u m s [i] + k$ 的值来查看是否有数对

class Solution {
public:
    int countKDifference(vector<int> &nums, int k) {
        // 统计 nums 的长度
        int len = nums.size();
        // 如果长度为 1 则不存在 i < j 可以直接返回 0
        if (len == 1) return 0;
        // 将结果初始化为 0
        int res = 0;
        // 创建一个哈希表 cnt 用来统计 nums 的数字
        unordered_map<int, int> cnt;
        // 遍历 nums 数组
        for (int num: nums) {
            // 判断是否存在 num - k 与 num + k 
            // 如果存在，则存在 cnt[num - k] 与 cnt[num + k] 组数对
            // 在结果 res 之中加上对应的值
            res += (cnt[num - k] ? cnt[num - k] : 0)+(cnt[num + k] ? cnt[num + k] : 0);
            // 统计 num 的数量
            cnt[num]++;
        }
        return res;
    }
};

class Solution {
    public int countKDifference(int[] nums, int k) {
        // 统计 nums 的长度
        int len = nums.length;
        // 如果长度为 1 则不存在 i < j 可以直接返回 0
        if (len == 1) return 0;
        // 将结果初始化为 0
        int res = 0;
        // 创建一个哈希表 cnt 用来统计 nums 的数字
        HashMap<Integer, Integer> cnt = new HashMap<Integer, Integer>();
        for (int num : nums) {
            // 判断是否存在 num - k 与 num + k 
            // 如果存在，则存在 cnt[num - k] 与 cnt[num + k] 组数对
            // 在结果 res 之中加上对应的值
            res += cnt.getOrDefault(num - k, 0) + cnt.getOrDefault(num + k, 0);
            // 统计 num 的数量
            cnt.put(num, cnt.getOrDefault(num, 0) + 1);
        }
        return res;
    }
}