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 n^2 n2)
-
空间复杂度O(1)
法二:一次遍历+哈希表
创建一个哈希表,统计每个数字出现的次数,
由于 ∣ n u m s [ i ] − n u m s [ j ] ∣ = k | nums[i] - nums[j] | = k ∣nums[i]−nums[j]∣=k可以拆解出$ nums[i] - nums[j] = k 与 与 与 nums[i] - nums[j] = -k $
因此可以转换为 n u m s [ j ] = n u m s [ i ] + k nums[j]=nums[i]+k nums[j]=nums[i]+k和 n u m s [ j ] = n u m s [ i ] − k nums[j]=nums[i]-k nums[j]=nums[i]−k
所以就可以通过查询 n u m s [ i ] + k nums[i]+k nums[i]+k与 n u m s [ i ] + k nums[i]+k nums[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;
}
}
-
时间复杂度O(n)
-
空间复杂度O(n)