fork download
  1. /* package whatever; // don't place package name! */
  2.  
  3. import java.util.*;
  4. import java.lang.*;
  5. import java.io.*;
  6.  
  7. /* Name of the class has to be "Main" only if the class is public. */
  8. class Ideone
  9. {
  10. public static void main (String[] args) throws java.lang.Exception
  11. {
  12. Scanner sc = new Scanner(System.in);
  13. int n = sc.nextInt();
  14. int arr []= new int[n+1];
  15.  
  16. for(int i=1;i<=n;i++){
  17. arr[i] = sc.nextInt();
  18. }
  19. int k = sc.nextInt();
  20. boolean ans = false;
  21. HashMap<Integer,Integer> map = new HashMap<>();
  22.  
  23. for(int i=1;i<=n;i++){
  24. if(!map.containsKey(arr[i])){
  25. map.put(arr[i],i);
  26. }else{
  27. if( i - map.get(arr[i]) <= k){
  28. ans = true;
  29. break;
  30. }
  31. map.put(arr[i],i);
  32. }
  33. }
  34.  
  35.  
  36. if(ans){
  37. System.out.println("YES");
  38. }else{
  39. System.out.println("NO");
  40. }
  41. }
  42. }
Success #stdin #stdout 0.13s 56468KB
stdin
5
3 2 3 3 1
2
stdout
YES