/* 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
int[] arr = {1, 7, 5, 9, 2, 12, 3};
int k = 2;
System.
out.
println(countPairsWithAbsoluteDiffK
(arr,k
));
}
static int countPairsWithAbsoluteDiffK(int[] arr, int k){
HashMap
<Integer,Integer
> map
= new HashMap
<>(); int cnt = 0;
for(int i=0;i<arr.length;i++){
int req1 = arr[i]+k;
int req2 = arr[i]-k;
if(map.containsKey(req1)){
cnt+=map.get(req1);
}
if(k!=0 && map.containsKey(req2)){
cnt+=map.get(req2);
}
map.put(arr[i],map.getOrDefault(arr[i],0)+1);
}
return cnt;
}
}
LyogcGFja2FnZSB3aGF0ZXZlcjsgLy8gZG9uJ3QgcGxhY2UgcGFja2FnZSBuYW1lISAqLwoKaW1wb3J0IGphdmEudXRpbC4qOwppbXBvcnQgamF2YS5sYW5nLio7CmltcG9ydCBqYXZhLmlvLio7CgovKiBOYW1lIG9mIHRoZSBjbGFzcyBoYXMgdG8gYmUgIk1haW4iIG9ubHkgaWYgdGhlIGNsYXNzIGlzIHB1YmxpYy4gKi8KY2xhc3MgSWRlb25lCnsKCXB1YmxpYyBzdGF0aWMgdm9pZCBtYWluIChTdHJpbmdbXSBhcmdzKSB0aHJvd3MgamF2YS5sYW5nLkV4Y2VwdGlvbgoJewoJCS8vIHlvdXIgY29kZSBnb2VzIGhlcmUKCQlpbnRbXSBhcnIgPSB7MSwgNywgNSwgOSwgMiwgMTIsIDN9OwogICAgICAgIGludCBrID0gMjsKCQlTeXN0ZW0ub3V0LnByaW50bG4oY291bnRQYWlyc1dpdGhBYnNvbHV0ZURpZmZLKGFycixrKSk7CgkJCgl9CgkKCXN0YXRpYyBpbnQgY291bnRQYWlyc1dpdGhBYnNvbHV0ZURpZmZLKGludFtdIGFyciwgaW50IGspewoJCUhhc2hNYXA8SW50ZWdlcixJbnRlZ2VyPiBtYXAgPSBuZXcgSGFzaE1hcDw+KCk7CgkJaW50IGNudCA9IDA7CgkJZm9yKGludCBpPTA7aTxhcnIubGVuZ3RoO2krKyl7CgkJCWludCByZXExID0gYXJyW2ldK2s7CgkJCWludCByZXEyID0gYXJyW2ldLWs7CgkJCWlmKG1hcC5jb250YWluc0tleShyZXExKSl7CgkJCQljbnQrPW1hcC5nZXQocmVxMSk7CgkJCX0KCQkJaWYoayE9MCAmJiBtYXAuY29udGFpbnNLZXkocmVxMikpewoJCQkJY250Kz1tYXAuZ2V0KHJlcTIpOwoJCQl9CgkJCW1hcC5wdXQoYXJyW2ldLG1hcC5nZXRPckRlZmF1bHQoYXJyW2ldLDApKzEpOwoJCX0KCQkKCQlyZXR1cm4gY250OwoJfQp9