#include <bits/stdc++.h>
using namespace std;
int count(vector<int>&arr , int k){
unordered_map<int,int>hmap;
int cnt=0;
for(int i=0;i<arr.size();i++){
if(hmap.find(k+arr[i])!=hmap.end()){
cnt+=hmap[k+arr[i]];
}
hmap[arr[i]]++;
}
return cnt;
}
int main(){
vector<int> arr={5,5,4,3,2,1};
int k=1;
cout<<count(arr,k);
}
I2luY2x1ZGUgPGJpdHMvc3RkYysrLmg+CnVzaW5nIG5hbWVzcGFjZSBzdGQ7CgppbnQgY291bnQodmVjdG9yPGludD4mYXJyICwgaW50IGspewogICAgdW5vcmRlcmVkX21hcDxpbnQsaW50PmhtYXA7CiAgICBpbnQgY250PTA7IAogICAgZm9yKGludCBpPTA7aTxhcnIuc2l6ZSgpO2krKyl7CiAgICAgICAKICAgICAgICAKICAgICAgICBpZihobWFwLmZpbmQoaythcnJbaV0pIT1obWFwLmVuZCgpKXsKICAgICAgICAgICAgY250Kz1obWFwW2srYXJyW2ldXTsKICAgICAgICB9CiAgICAgICAgaG1hcFthcnJbaV1dKys7CgogICAgfQpyZXR1cm4gY250OyAgICAKfQppbnQgbWFpbigpewogICAgdmVjdG9yPGludD4gYXJyPXs1LDUsNCwzLDIsMX07CiAgICBpbnQgaz0xOwoKICAgIGNvdXQ8PGNvdW50KGFycixrKTsKCn0=