/* package whatever; // don't place package name! */
import java.util.*;
import java.lang.*;
import java.io.*;
/* Name of the class has to be "Main" only if the class is public. */
class Ideone
{
{
// your code goes here
}
}
class Solution {
public int countKDifference(int[] nums, int k) {
HashMap
<Integer , Integer
> map
= new HashMap
<>(); int n = nums.length;
int pairs = 0;
for(int i = 0 ; i< n ; i++){
int curr = nums[i];
int req1 = curr - k;
int req2 = curr + k;
int type1 = 1*map.getOrDefault(req1 , 0);
int type2 = 1*map.getOrDefault(req2 , 0);
pairs+=(type1 + type2);
map.put(curr , map.getOrDefault(curr , 0) + 1);
}
return pairs;
}
}
LyogcGFja2FnZSB3aGF0ZXZlcjsgLy8gZG9uJ3QgcGxhY2UgcGFja2FnZSBuYW1lISAqLwoKaW1wb3J0IGphdmEudXRpbC4qOwppbXBvcnQgamF2YS5sYW5nLio7CmltcG9ydCBqYXZhLmlvLio7CgovKiBOYW1lIG9mIHRoZSBjbGFzcyBoYXMgdG8gYmUgIk1haW4iIG9ubHkgaWYgdGhlIGNsYXNzIGlzIHB1YmxpYy4gKi8KY2xhc3MgSWRlb25lCnsKCXB1YmxpYyBzdGF0aWMgdm9pZCBtYWluIChTdHJpbmdbXSBhcmdzKSB0aHJvd3MgamF2YS5sYW5nLkV4Y2VwdGlvbgoJewoJCS8vIHlvdXIgY29kZSBnb2VzIGhlcmUKCX0KfQoKY2xhc3MgU29sdXRpb24gewogICAgcHVibGljIGludCBjb3VudEtEaWZmZXJlbmNlKGludFtdIG51bXMsIGludCBrKSB7CiAgICAgICAgSGFzaE1hcDxJbnRlZ2VyICwgSW50ZWdlcj4gbWFwID0gIG5ldyBIYXNoTWFwPD4oKTsKICAgICAgICBpbnQgbiA9IG51bXMubGVuZ3RoOwogICAgICAgIGludCBwYWlycyA9IDA7CiAgICAgICAgZm9yKGludCBpICA9IDAgOyBpPCBuIDsgaSsrKXsKICAgICAgICAgICAgaW50IGN1cnIgPSBudW1zW2ldOwogICAgICAgICAgICBpbnQgcmVxMSA9IGN1cnIgLSBrOwogICAgICAgICAgICBpbnQgcmVxMiA9IGN1cnIgKyBrOwogICAgICAgICAgICBpbnQgdHlwZTEgPSAxKm1hcC5nZXRPckRlZmF1bHQocmVxMSAsIDApOwogICAgICAgICAgICBpbnQgdHlwZTIgPSAxKm1hcC5nZXRPckRlZmF1bHQocmVxMiAsIDApOwogICAgICAgICAgICBwYWlycys9KHR5cGUxICsgdHlwZTIpOwogICAgICAgICAgICBtYXAucHV0KGN1cnIgLCBtYXAuZ2V0T3JEZWZhdWx0KGN1cnIgLCAwKSArIDEpOyAKICAgICAgICB9CiAgICAgICAgcmV0dXJuIHBhaXJzOwogICAgfQp9