Skip to content
Open
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
Update binary_tree_part_2.py
Adding comments to the delete method, as it took me longer than expected to understand the code. Even after I went through the video.
  • Loading branch information
avigod authored Jun 30, 2024
commit c0523db9c2ffeedb665c76b17d21ca84bcca6d4e
17 changes: 10 additions & 7 deletions data_structures/9_Binary_Tree_2/binary_tree_part_2.py
Original file line number Diff line number Diff line change
Expand Up @@ -49,24 +49,27 @@ def in_order_traversal(self):
return elements

def delete(self, val):
if val < self.data:
#traverse till you find the node to delete
if val < self.data: #then go left
if self.left:
self.left = self.left.delete(val)
elif val > self.data:
elif val > self.data: #then go right
if self.right:
self.right = self.right.delete(val)
else:
if self.left is None and self.right is None:
else:#found node to delete i.e. val = self.data
if self.left is None and self.right is None: #case of deleting leaf node
return None
elif self.left is None:
elif self.left is None: #case of deleting node with one child
return self.right
elif self.right is None:
elif self.right is None: #case of deleting node with one child
return self.left

#case of deleting node with two children
min_val = self.right.find_min()
self.data = min_val
self.right = self.right.delete(min_val)


#if not leaf, then you have to return self
return self

def find_max(self):
Expand Down