Advertisement
Guest User

Untitled

a guest
Feb 24th, 2021
51
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
Python 0.58 KB | None | 0 0
  1. n, k = [int(i) for i in input().split()]
  2. players = [int(i) for i in input().split()]
  3.  
  4. players_top = list(enumerate(players))
  5.  
  6. best = (0, -1000000000000)
  7.  
  8. end = -k + 1
  9. if end == 0:
  10.     end = n + 1
  11.  
  12. for player in sorted(players_top, key=lambda x:x[0])[k - 1: end]:
  13.     value = player[1] - max([9999999] if len(players[player[0] + 1:]) == 0 else players[player[0] + 1:])
  14.     best_id, best_value = best
  15.  
  16.     if value > best_value:
  17.         best = player
  18.  
  19. left = sorted(players[:best[0] + 1])[::-1]
  20. right = sorted(players[best[0] + 1:])
  21.  
  22. print(sum(left[:k]) - sum(right[:k]))
  23.  
  24.  
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement