import java.util.HashMap;
import java.util.Map;
public class Main {
public static void main
(String[] args
) { int[] b = {1, 5, 3, 4, 2};
int k = 2;
int count = 0;
Map
<Integer, Integer
> freqMap
= new HashMap
<>(); for (int j = 0; j < b.length; ++j) {
int target = b[j] + k;
if (freqMap.containsKey(target)) {
count += freqMap.get(target);
}
freqMap.put(b[j], freqMap.getOrDefault(b[j], 0) + 1);
}
}
}
aW1wb3J0IGphdmEudXRpbC5IYXNoTWFwOwppbXBvcnQgamF2YS51dGlsLk1hcDsKCnB1YmxpYyBjbGFzcyBNYWluIHsKICAgCiAgICBwdWJsaWMgc3RhdGljIHZvaWQgbWFpbihTdHJpbmdbXSBhcmdzKSB7CiAgICAgICAgaW50W10gYiA9IHsxLCA1LCAzLCA0LCAyfTsKICAgICAgICBpbnQgayA9IDI7CiAgICAgICAgIGludCBjb3VudCA9IDA7CiAgICAgICAgTWFwPEludGVnZXIsIEludGVnZXI+IGZyZXFNYXAgPSBuZXcgSGFzaE1hcDw+KCk7CiAgICAgICAgZm9yIChpbnQgaiA9IDA7IGogPCBiLmxlbmd0aDsgKytqKSB7CiAgICAgICAgICAgIGludCB0YXJnZXQgPSBiW2pdICsgazsKICAgICAgICAgICAgaWYgKGZyZXFNYXAuY29udGFpbnNLZXkodGFyZ2V0KSkgewogICAgICAgICAgICAgICAgY291bnQgKz0gZnJlcU1hcC5nZXQodGFyZ2V0KTsKICAgICAgICAgICAgfQogICAgICAgICAgICBmcmVxTWFwLnB1dChiW2pdLCBmcmVxTWFwLmdldE9yRGVmYXVsdChiW2pdLCAwKSArIDEpOwogICAgICAgIH0KICAgICAgICBTeXN0ZW0ub3V0LnByaW50KGNvdW50KTsKICAgICAgICAKICAgIH0KfQo=