/* 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 = {3,2,1,5,4};
int k =2;
System.
out.
println("Number Of pairs with absolute diff k are " + cntPairs
(arr,k
)); }
public static int cntPairs(int[] arr, int k){
HashMap
<Integer,Integer
> map
= new HashMap
<>(); int res = 0;
for(int i=0;i<arr.length;i++){
if(map.containsKey(arr[i]-k)){
res+=map.get(arr[i]-k);
}
if(map.containsKey(arr[i]+k)){
res+=map.get(arr[i]+k);
}
map.put(arr[i],map.getOrDefault(arr[i],0)+1);
}
return res;
}
}
LyogcGFja2FnZSB3aGF0ZXZlcjsgLy8gZG9uJ3QgcGxhY2UgcGFja2FnZSBuYW1lISAqLwoKaW1wb3J0IGphdmEudXRpbC4qOwppbXBvcnQgamF2YS5sYW5nLio7CmltcG9ydCBqYXZhLmlvLio7CgovKiBOYW1lIG9mIHRoZSBjbGFzcyBoYXMgdG8gYmUgIk1haW4iIG9ubHkgaWYgdGhlIGNsYXNzIGlzIHB1YmxpYy4gKi8KY2xhc3MgSWRlb25lCnsKCXB1YmxpYyBzdGF0aWMgdm9pZCBtYWluIChTdHJpbmdbXSBhcmdzKSB0aHJvd3MgamF2YS5sYW5nLkV4Y2VwdGlvbgoJewoJCS8vIHlvdXIgY29kZSBnb2VzIGhlcmUKCQlpbnRbXSBhcnIgPSB7MywyLDEsNSw0fTsKCQlpbnQgayA9MjsKCQlTeXN0ZW0ub3V0LnByaW50bG4oIk51bWJlciBPZiBwYWlycyB3aXRoIGFic29sdXRlIGRpZmYgayBhcmUgIiArIGNudFBhaXJzKGFycixrKSk7Cgl9CglwdWJsaWMgc3RhdGljIGludCBjbnRQYWlycyhpbnRbXSBhcnIsIGludCBrKXsKCQlIYXNoTWFwPEludGVnZXIsSW50ZWdlcj4gbWFwID0gbmV3IEhhc2hNYXA8PigpOwoJCWludCByZXMgPSAwOwoJCWZvcihpbnQgaT0wO2k8YXJyLmxlbmd0aDtpKyspewoJCQlpZihtYXAuY29udGFpbnNLZXkoYXJyW2ldLWspKXsKCQkJCXJlcys9bWFwLmdldChhcnJbaV0tayk7CgkJCX0KCQkJaWYobWFwLmNvbnRhaW5zS2V5KGFycltpXStrKSl7CgkJCQlyZXMrPW1hcC5nZXQoYXJyW2ldK2spOwoJCQl9CgkJCW1hcC5wdXQoYXJyW2ldLG1hcC5nZXRPckRlZmF1bHQoYXJyW2ldLDApKzEpOwoJCX0KCQlyZXR1cm4gcmVzOwoJfQp9