fork download
  1. #include <bits/stdc++.h>
  2. using namespace std;
  3.  
  4. int ans(vector<int> &v,int &k){
  5. unordered_map<int,int>m;
  6. int count=0;
  7. for(int i=0;i<v.size();i++){
  8. int c=k+v[i];int c1=v[i]-k;
  9. if(m.find(c)!=m.end()){
  10. count+=m[c];
  11. }
  12. if(k!=0 && (m.find(c1)!=m.end()))count+=m[c1];
  13. m[v[i]]++;
  14. }
  15. return count;
  16. }
  17. int main() {
  18. // your code goes here
  19. int n;cin>>n;
  20. vector<int>v(n);
  21. for(int i=0;i<n;i++)cin>>v[i];
  22. int k;cin>>k;
  23. cout<<ans(v,k)<<endl;
  24. return 0;
  25. }
Success #stdin #stdout 0s 5320KB
stdin
5
1 5 3 4 2
2
stdout
3