Advertisement
Purposelessness

Untitled

Feb 2nd, 2023
25
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
text 0.78 KB | None | 0 0
  1. def heapify(arr, n, i):
  2. largest = i
  3. left = 2 * i + 1
  4. right = 2 * i + 2
  5.  
  6. if left < n and arr[left] > arr[largest]:
  7. largest = left
  8.  
  9. if right < n and arr[right] > arr[largest]:
  10. largest = right
  11.  
  12. if largest != i:
  13. arr[i], arr[largest] = arr[largest], arr[i]
  14. heapify(arr, n, largest)
  15.  
  16.  
  17. def heap_sort(arr, n):
  18. for i in range(int(n / 2) - 1, -1, -1):
  19. heapify(arr, n, i)
  20.  
  21.  
  22. def main():
  23. _result = []
  24. _input = list(map(int, input().split()))
  25. _delete = list(map(int, input().split()))
  26. heap_sort(_input, len(_input))
  27. for i in _delete:
  28. _input[i] = _input[-1]
  29. _input.pop()
  30. heapify(_input, len(_input), i)
  31. print(_input)
  32.  
  33.  
  34. if __name__ == "__main__":
  35. main()
  36.  
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement