Skip to content

Commit

Permalink
Create min-bintree-diff.py
Browse files Browse the repository at this point in the history
janky-ahh solution to find the minimum difference between nodes in a binary search tree, could be very easily optimised but im not gonna right now
  • Loading branch information
gabedonnan authored Feb 17, 2023
1 parent 70efb2c commit aef51cb
Showing 1 changed file with 25 additions and 0 deletions.
25 changes: 25 additions & 0 deletions min-bintree-diff.py
Original file line number Diff line number Diff line change
@@ -0,0 +1,25 @@
# Definition for a binary tree node.
# class TreeNode:
# def __init__(self, val=0, left=None, right=None):
# self.val = val
# self.left = left
# self.right = right
class Solution:
def minDiffInBST(self, root_node: Optional[TreeNode]) -> int:

def search(root: Optional[TreeNode], prevs: List[int]) -> int:
if len(prevs) > 0:
prevdiff = min([abs(root.val - prev) for prev in prevs])
else:
prevdiff = math.inf
#print(root.val, prevs, prevdiff)
if root.left == None and root.right == None:
return prevdiff
elif root.left == None and root.right != None:
return min([abs(root.val - root.right.val), search(root.right, prevs + [root.val]), prevdiff])
elif root.right == None and root.left != None:
return min([abs(root.val - root.left.val), search(root.left, prevs + [root.val]), prevdiff])
else:
return min([search(root.left, prevs + [root.val]), search(root.right, prevs + [root.val]), abs(root.val - root.left.val), abs(root.val - root.right.val), prevdiff])
return search(root_node, [])

0 comments on commit aef51cb

Please sign in to comment.