Create heap_sort.py

This commit is contained in:
marina 2023-08-02 13:32:18 -07:00 committed by GitHub
parent 4caa1fe7f4
commit f7779dc6e3
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

27
heaps/heap_sort.py Normal file
View File

@ -0,0 +1,27 @@
#!/usr/bin/env python3
# -*- coding: utf-8 -*-
# author: bt3gl
def heap_sort(self, lst: List[int]) -> None:
def max_heapify(heap_size, index):
left, right = 2 * index + 1, 2 * index + 2
largest = index
if left < heap_size and lst[left] > lst[largest]:
largest = left
if right < heap_size and lst[right] > lst[largest]:
largest = right
if largest != index:
lst[index], lst[largest] = lst[largest], lst[index]
max_heapify(heap_size, largest)
for i in range(len(lst) // 2 - 1, -1, -1):
max_heapify(len(lst), i)
for i in range(len(lst) - 1, 0, -1):
lst[i], lst[0] = lst[0], lst[i]
max_heapify(i, 0)