-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathhsort.py
38 lines (27 loc) · 782 Bytes
/
hsort.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
def heapify(arr, n, i):
# Find largest among root and children
largest = i
l = 2 * i + 1
r = 2 * i + 2
if l < n and arr[i] < arr[l]:
largest = l
if r < n and arr[largest] < arr[r]:
largest = r
# If root is not largest, swap with largest and continue heapifying
if largest != i:
arr[i],arr[largest] = arr[largest],arr[i]
heapify(arr, n, largest)
def heapSort(arr):
n = len(arr)
# Build max heap
for i in range(n, 0, -1):
heapify(arr, n, i)
for i in range(n-1, 0, -1):
# swap
arr[i], arr[0] = arr[0], arr[i]
#heapify root element
heapify(arr, i, 0)
arr = [ 12, 11, 13, 5, 6, 7]
heapSort(arr)
print ("Sorted array is : ")
print(arr)