Easy
Count Number of Pairs With Absolute Difference K — C++
Full explanation · Time O(n) · Space O(n)
// Time: O(n)
// Space: O(n)
class Solution {
public:
int countKDifference(vector<int>& nums, int k) {
unordered_map<int, int> lookup;
int result = 0;
for (const auto& x : nums) {
if (lookup.count(x - k)) {
result += lookup[x - k];
}
if (lookup.count(x + k)) {
result += lookup[x + k];
}
++lookup[x];
}
return result;
}
};