fork download
  1. #include <bits/stdc++.h>
  2. using namespace std;
  3.  
  4. bool fun(vector<int> nums, int k) {
  5. unordered_map<int, int> mpp;
  6.  
  7. for (int i = 0; i < nums.size(); i++) {
  8.  
  9. if (mpp.find(nums[i]) != mpp.end()) {
  10. int lastIndex = mpp[nums[i]];
  11. int distance = i - lastIndex;
  12.  
  13. if (distance <= k)
  14. return true;
  15. }
  16.  
  17. mpp[nums[i]] = i;
  18. }
  19.  
  20. return false;
  21. }
  22.  
  23. int main() {
  24. vector<int> nums = {1, 2, 2, 1, 2, 3};
  25. int k = 2;
  26.  
  27. cout << fun(nums, k);
  28.  
  29. return 0;
  30. }
Success #stdin #stdout 0.01s 5288KB
stdin
Standard input is empty
stdout
1