2000. Reverse Prefix of Word #
题目 #
- 给定整数数组
nums和一个整数k。 - 返回数对
(i, j)的数目，满足i < j且|nums[i] - nums[j]| == k。 1 <= nums[i] <= 1001 <= k <= 99
思路 #
哈希 #
代码 #
哈希 #
class Solution {
public int countKDifference(int[] nums, int k) {
Map<Integer, Integer> map = new HashMap<>();
int ans = 0;
for (int num: nums) {
ans += map.getOrDefault(num - k, 0);
ans += map.getOrDefault(num + k, 0);
map.put(num, map.getOrDefault(num, 0) + 1);
}
return ans;
}
}