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

链接

力扣https://leetcode-cn.com/problems/count-number-of-pairs-with-absolute-difference-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]

提示

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

思路一 暴力

直接双层for循环暴力求解

C++ Code

class Solution {
public:
    int countKDifference(vector<int>& nums, int k) {
        int count=0;
        for(int i=0;i<nums.size()-1;i++)
            for(int j=i+1;j<nums.size();j++)
                if(abs(nums[i]-nums[j])==k)
                    count++;
        return count;
    }
};

复杂度

时间复杂度O(n²)    空间复杂度O(1)

思路二 哈希表

遍历到每个数nums[i]时，将该数存入哈希表当中（个数+1），同时统计哈希表里的nums[j]的个数，这样就是一堆符合题目要求的数对。

C++ Code

class Solution {
public:
    int countKDifference(vector<int>& nums, int k) {
        int count = 0;
        unordered_map<int, int> cnt;
        for (int i = 0; i < nums.size(); i++) {
            count += (cnt.count(nums[i] - k) ? cnt[nums[i] - k] : 0);
            count += (cnt.count(nums[i] + k) ? cnt[nums[i] + k] : 0);
            ++cnt[nums[i]];
        }
        return count;
    }
};

复杂度

时间复杂度O(n)    空间复杂度O(n)