/* 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
{
	public static void main (String[] args) throws java.lang.Exception
	{
		// your code goes here
 
		int[] arr = {1, 5, 3, 4, 2};
        int k = 2;
		System.out.println(countAllPairsWithDiffK(arr,k));
	}
	static int countAllPairsWithDiffK(int[] arr, int k){
		HashMap<Integer,Integer> map = new HashMap<>();
		int cnt = 0;
		for(int i=0;i<arr.length;i++){
			if(map.containsKey(k+arr[i])){
				cnt+=map.get(k+arr[i]);
			}
			map.put(arr[i],map.getOrDefault(arr[i],0)+1);
		}
		return cnt;
	}
}

LyogcGFja2FnZSB3aGF0ZXZlcjsgLy8gZG9uJ3QgcGxhY2UgcGFja2FnZSBuYW1lISAqLwoKaW1wb3J0IGphdmEudXRpbC4qOwppbXBvcnQgamF2YS5sYW5nLio7CmltcG9ydCBqYXZhLmlvLio7CgovKiBOYW1lIG9mIHRoZSBjbGFzcyBoYXMgdG8gYmUgIk1haW4iIG9ubHkgaWYgdGhlIGNsYXNzIGlzIHB1YmxpYy4gKi8KY2xhc3MgSWRlb25lCnsKCXB1YmxpYyBzdGF0aWMgdm9pZCBtYWluIChTdHJpbmdbXSBhcmdzKSB0aHJvd3MgamF2YS5sYW5nLkV4Y2VwdGlvbgoJewoJCS8vIHlvdXIgY29kZSBnb2VzIGhlcmUKCQkKCQlpbnRbXSBhcnIgPSB7MSwgNSwgMywgNCwgMn07CiAgICAgICAgaW50IGsgPSAyOwoJCVN5c3RlbS5vdXQucHJpbnRsbihjb3VudEFsbFBhaXJzV2l0aERpZmZLKGFycixrKSk7Cgl9CglzdGF0aWMgaW50IGNvdW50QWxsUGFpcnNXaXRoRGlmZksoaW50W10gYXJyLCBpbnQgayl7CgkJSGFzaE1hcDxJbnRlZ2VyLEludGVnZXI+IG1hcCA9IG5ldyBIYXNoTWFwPD4oKTsKCQlpbnQgY250ID0gMDsKCQlmb3IoaW50IGk9MDtpPGFyci5sZW5ndGg7aSsrKXsKCQkJaWYobWFwLmNvbnRhaW5zS2V5KGsrYXJyW2ldKSl7CgkJCQljbnQrPW1hcC5nZXQoaythcnJbaV0pOwoJCQl9CgkJCW1hcC5wdXQoYXJyW2ldLG1hcC5nZXRPckRlZmF1bHQoYXJyW2ldLDApKzEpOwoJCX0KCQlyZXR1cm4gY250OwoJfQp9