mirror of
https://github.com/TheAlgorithms/Python.git
synced 2025-01-31 06:33:44 +00:00
Merge pull request #18 from arijit-pande/master
added implementation for heap sort algorithm
This commit is contained in:
commit
9b9e4cdc7c
39
sorts/heap_sort.py
Normal file
39
sorts/heap_sort.py
Normal file
|
@ -0,0 +1,39 @@
|
||||||
|
|
||||||
|
from __future__ import print_function
|
||||||
|
|
||||||
|
|
||||||
|
def heapify(unsorted,index,heap_size):
|
||||||
|
largest = index
|
||||||
|
left_index = 2*index + 1
|
||||||
|
right_index = 2*index + 2
|
||||||
|
if left_index < heap_size and unsorted[left_index] > unsorted[largest]:
|
||||||
|
largest = left_index
|
||||||
|
|
||||||
|
if right_index < heap_size and unsorted[right_index] > unsorted[largest]:
|
||||||
|
largest = right_index
|
||||||
|
|
||||||
|
if largest != index:
|
||||||
|
unsorted[largest],unsorted[index] = unsorted[index],unsorted[largest]
|
||||||
|
heapify(unsorted,largest,heap_size)
|
||||||
|
|
||||||
|
def heap_sort(unsorted):
|
||||||
|
n=len(unsorted)
|
||||||
|
for i in range (n/2 - 1 , -1, -1) :
|
||||||
|
heapify(unsorted,i,n)
|
||||||
|
for i in range(n - 1, -1, -1):
|
||||||
|
unsorted[0], unsorted[i] = unsorted[i], unsorted[0]
|
||||||
|
heapify(unsorted,0,i)
|
||||||
|
return unsorted
|
||||||
|
|
||||||
|
|
||||||
|
if __name__ == '__main__':
|
||||||
|
import sys
|
||||||
|
if sys.version_info.major < 3:
|
||||||
|
input_function = raw_input
|
||||||
|
else:
|
||||||
|
input_function = input
|
||||||
|
|
||||||
|
user_input = input_function('Enter numbers separated by coma:\n')
|
||||||
|
unsorted = [int(item) for item in user_input.split(',')]
|
||||||
|
print (heap_sort(unsorted))
|
||||||
|
|
Loading…
Reference in New Issue
Block a user