fork download 
  1. #include <bits/stdc++.h>
  2. using namespace std;
  3.  
  4. int countKDifference(vector<int>& nums, int k) {
  5.     int n = nums.size();
  6.     unordered_map<int, int> freq;
  7.     for (int i : nums){
  8.     	freq[i]++;
  9.     }
  10.     int count = 0;
  11.     for (int i : nums){
  12.     	if(freq[i+k] > 0){
  13.     		count += (freq[i] * freq[i+k]);
  14.     	}
  15.     	if(freq[i-k] > 0){
  16.     		count += (freq[i] * freq[i-k]);
  17.     	}
  18.  
  19.     	freq[i] = 0;
  20.     }
  21.  
  22.     return count;
  23. }
  24.  
  25. int main() {
  26. 	// your code goes here
  27. 	vector<int> nums = {1, 3};
  28. 	cout << countKDifference(nums, 3);
  29. 	return 0;
  30. }
Success #stdin #stdout 0.01s 5288KB
comments (?)
stdin 
Standard input is empty
stdout 
Standard output is empty
https://ideone.com/yXRLuw
language:
C++14 (gcc 8.3)
created:
21 days ago
visibility:
public

Share or Embed source code

Discover > Sphere Engine API

The brand new service which powers Ideone!

Discover > IDE Widget

Widget for compiling and running the source code in a web browser!