fork download
  1. def main():
  2. n = int(input())
  3. k = int(input())
  4. arr = [int(input()) for _ in range (n)]
  5.  
  6. mp = {}
  7. for i in range(n):
  8. if arr[i] in mp and i-mp[arr[i]]<=k:
  9. print('yes')
  10. return
  11. mp[arr[i]]= i
  12. print('no')
  13. return
  14.  
  15. if __name__=="__main__":
  16. main()
Success #stdin #stdout 0.11s 14084KB
stdin
6
3
1
2
3
1
4
5
stdout
yes