fork download
  1. #include <bits/stdc++.h>
  2. using namespace std;
  3.  
  4. int main() {
  5. // your code goes here
  6. int arr[]={1,2,3,1,4,5};
  7. int n=sizeof(arr)/sizeof(arr[0]);
  8. int k=3;
  9. unordered_map<int,int>u;
  10. for(int i=0;i<n;i++){
  11. if(u.find(arr[i])!=u.end()){
  12. int dist=i-u[arr[i]];
  13. if(dist<=k){
  14. cout<<"Yes";
  15. u[arr[i]]=i;
  16. return 0;
  17. }
  18.  
  19. }
  20. else{
  21. u[arr[i]]=i;
  22. }
  23.  
  24. }
  25. cout<<"no";
  26. return 0;
  27. }
Success #stdin #stdout 0s 5320KB
stdin
Standard input is empty
stdout
Yes