1.leetcode题目 235. 二叉搜索树的最近公共祖先 - 力扣(LeetCode)
解决:
①用昨天做二叉树的最近公共祖先的方法做:
# Definition for a binary tree node.
# class TreeNode:
# def __init__(self, x):
# self.val = x
# self.left = None
# self.right = None
class Solution:
def lowestCommonAncestor(self, root: 'TreeNode', p: 'TreeNode', q: 'TreeNode') -> 'TreeNode':
if root==p or root==q or root is None:
return root
left = self.lowestCommonAncestor(root.left,p,q)
right = self.lowestCommonAncestor(root.right,p,q)
if left is not None and right is not None:
return root
if left is None:return right
return left
②利用二叉搜索树的有序性:
# Definition for a binary tree node.
# class TreeNode:
# def __init__(self, x):
# self.val = x
# self.left = None
# self.right = None
class Solution:
def lowestCommonAncestor(self, root: 'TreeNode', p: 'TreeNode', q: 'TreeNode') -> 'TreeNode':
if root.val>p.val and root.val>q.val:
return self.lowestCommonAncestor(root.left,p,q)
if root.val<p.val and root.val<q.val:
return self.lowestCommonAncestor(root.right,p,q)
return root
2.leetcode题目 701. 二叉搜索树中的插入操作 - 力扣(LeetCode)
解决:
# 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 insertIntoBST(self, root: Optional[TreeNode], val: int) -> Optional[TreeNode]:
if root is None:
node = TreeNode(val)
return node
if root.val>val:
root.left = self.insertIntoBST(root.left,val)
if root.val<val:
root.right = self.insertIntoBST(root.right,val)
return root
3.leetcode题目 450. 删除二叉搜索树中的节点 - 力扣(LeetCode)
解决:这题需要充分考虑被删除节点的左右节点的安置情况
# 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 deleteNode(self, root: Optional[TreeNode], key: int) -> Optional[TreeNode]:
if root is None:
return root
if root.val == key:
if root.left is None and root.right is None:
return None
elif root.left is None:
return root.right
elif root.right is None:
return root.left
else:
cur = root.right
while cur.left is not None:
cur = cur.left
cur.left = root.left
return root.right
if root.val > key:
root.left = self.deleteNode(root.left,key)
if root.val < key:
root.right = self.deleteNode(root.right,key)
return root