mathdatech/Python
1
0
Fork 0
mirror of https://github.com/TheAlgorithms/Python.git synced 2025-01-18 16:27:02 +00:00
Code Issues Packages Projects Releases Wiki Activity

Update heap.py (#726)

Browse Source 
Added comments for the better understanding of heap.
This commit is contained in:
Akash Ali 2019-03-04 00:49:36 -08:00 committed by John Law
parent 88b6caa30a
commit 6f6510623c
1 changed files with 7 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
data_structures/heap/heap.py
View File

@ -7,8 +7,9 @@ try:
except NameError: except NameError:
raw_input = input # Python 3 raw_input = input # Python 3
#This heap class start from here.
class Heap: class Heap:
def __init__(self): def __init__(self): #Default constructor of heap class.
self.h = [] self.h = []
self.currsize = 0 self.currsize = 0
@ -37,13 +38,13 @@ class Heap:
self.h[m] = temp self.h[m] = temp
self.maxHeapify(m) self.maxHeapify(m)
def buildHeap(self,a): def buildHeap(self,a): #This function is used to build the heap from the data container 'a'.
self.currsize = len(a) self.currsize = len(a)
self.h = list(a) self.h = list(a)
for i in range(self.currsize//2,-1,-1): for i in range(self.currsize//2,-1,-1):
self.maxHeapify(i) self.maxHeapify(i)
def getMax(self): def getMax(self): #This function is used to get maximum value from the heap.
if self.currsize >= 1: if self.currsize >= 1:
me = self.h[0] me = self.h[0]
temp = self.h[0] temp = self.h[0]
@ -54,7 +55,7 @@ class Heap:
return me return me
return None return None
def heapSort(self): def heapSort(self): #This function is used to sort the heap.
size = self.currsize size = self.currsize
while self.currsize-1 >= 0: while self.currsize-1 >= 0:
temp = self.h[0] temp = self.h[0]
@ -64,7 +65,7 @@ class Heap:
self.maxHeapify(0) self.maxHeapify(0)
self.currsize = size self.currsize = size
def insert(self,data): def insert(self,data): #This function is used to insert data in the heap.
self.h.append(data) self.h.append(data)
curr = self.currsize curr = self.currsize
self.currsize+=1 self.currsize+=1
@ -74,7 +75,7 @@ class Heap:
self.h[curr] = temp self.h[curr] = temp
curr = curr/2 curr = curr/2
def display(self): def display(self): #This function is used to print the heap.
print(self.h) print(self.h)
def main(): def main():