哈希表 -leetcode-2006. 差的绝对值为 K 的数对数目

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

提示：

• $1<=nums.length<=200$
• $1<=nums[i]<=100$
• $1<=k<=99$

暴力

数据范围比较小，可以直接 2层for暴力枚举

这里省略代码

//...

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

哈希表

思路：🤔

• 题目要求 i < j，所以可以采用 “边遍历，边记录”的方法，使用 哈希表 记录 当前元素之前出现的元素的个数，即 $arr[i]$
• 要求 $|nums[i]-nums[j]|==k$，令 $cnt=nums[i]$，即求当前元素“之前” $cnt-k$ 和 $cnt+k$ 出现的次数

class Solution {
    public int countKDifference(int[] nums, int k) {
        int res = 0;
        int[] arr = new int[101]; // 最大范围：[1, 100]
        for (int i = 0; i < nums.length; i++) {
            int cnt = nums[i];
            // if 的作用是：防止arr数组下标越界
            if (cnt - k > 0) res += arr[cnt - k];
            if (cnt + k <= 100) res += arr[cnt + k];
            // 边遍历，边记录（因为题目保证了 i < j）
            arr[cnt]++;
        }
        return res;
    }
}

用空间换时间，hash表浪费了空间，但是提高了时间效率

• 时间复杂度：$O(n)$
• 空间复杂度：$O(n)$