fork download
  1. #include <bits/stdc++.h>
  2. using namespace std;
  3.  
  4. int main() {
  5.  
  6. int n,k;
  7. cin>>n>>k;
  8.  
  9. vector<int> numbers(n);
  10.  
  11. for(auto& number: numbers){
  12. cin>>number;
  13. }
  14.  
  15. unordered_map<int,int> lastIndex;
  16.  
  17. bool answer = false;
  18.  
  19. for(int i=0; i<n; i++){
  20. if(lastIndex.find(numbers[i]) != lastIndex.end()){
  21. if( i - lastIndex[numbers[i]] <= k ){
  22. answer = true;
  23. break;
  24. }
  25.  
  26.  
  27. }
  28.  
  29. lastIndex[numbers[i]] = i;
  30.  
  31. }
  32.  
  33. (answer == true)?(cout<<"yes"):(cout<<"no");
  34.  
  35. return 0;
  36. }
Success #stdin #stdout 0s 5316KB
stdin
5 2
3 2 4 3 1
stdout
no