From f7779dc6e3744acd82624f223ee132a7ee1a1e4c Mon Sep 17 00:00:00 2001 From: marina <138340846+bt3gl-cryptographer@users.noreply.github.com> Date: Wed, 2 Aug 2023 13:32:18 -0700 Subject: [PATCH] Create heap_sort.py --- heaps/heap_sort.py | 27 +++++++++++++++++++++++++++ 1 file changed, 27 insertions(+) create mode 100644 heaps/heap_sort.py diff --git a/heaps/heap_sort.py b/heaps/heap_sort.py new file mode 100644 index 0000000..081bf89 --- /dev/null +++ b/heaps/heap_sort.py @@ -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) +