文章目录
题目描述
给你一个整数数组 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
题解思路
方法一:暴力
我们可以使用两层循环,一层遍历 i,一层遍历 j,对每个(i,j) 的组合,判断差的绝对值是否为 k,统计所有符合条件的数对。
class Solution {
public:
int countKDifference(vector<int>& nums, int k) {
int res = 0;
for (int i = 0; i < nums.size(); ++i) {
for (int j = i + 1; j < nums.size(); ++j) {
if (abs(nums[i] - nums[j]) == k) {
++res;
}
}
}
return res;
}
};
复杂度分析
-
时间复杂度: O ( n 2 ) O(n^2) O(n2),其中 n 为数组 nums 的长度。我们使用了两层循环来寻找所有符合条件的数对。
-
空间复杂度: O ( 1 ) O(1) O(1)。我们仅使用常数空间。