给你一个整数数组 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 的数对为:
输入:nums = [1,3], k = 3
输出:0
解释:没有任何数对差的绝对值为 3 。
示例 3:
输入:nums = [3,2,1,5,4], k = 2
输出:3
解释:差的绝对值为 2 的数对为:
提示:
1 <= nums.length <= 200
1 <= nums[i] <= 100
1 <= k <= 99
直接对nums遍历,每次循环从nums[i]处开始查找满足|nums[i] - nums[j]| = |k|的元素个数。
int countDifference(vector<int> &nums, int k)
{
int result = 0;
for(int i = 0; i < nums.size() - 1; i++){
int plus = count((nums.begin() + i), nums.end(), nums[i] + k);
int minus = count((nums.begin() + i), nums.end(), nums[i] - k);
result += (plus + minus);
}
return result;
}