235. 二叉搜索树的最近公共祖先
本题思路:
还是要利用二叉搜索树的特性,中序遍历为有序数组。如果pq两个节点都小于root,那么最近公共祖肯定是在他的左子树,如果都大于那么,肯定就在右子树。然后直接return root
代码:
def lowestCommonAncestor(self, root: 'TreeNode', p: 'TreeNode', q: 'TreeNode') -> 'TreeNode':
if p.val<root.val and q.val<root.val:
return self.lowestCommonAncestor(root.left,p,q)
if p.val>root.val and q.val>root.val:
return self.lowestCommonAncestor(root.right,p,q)
return root
701. 二叉搜索树中的插入操作
本题目涉及到一个二叉搜索树树的插入操作,如果当root等于none的时候,实际上我们这里就要开始插入操作了,因为我们对二叉搜索树的特性有了解,如果这个val<根节点的话,它一定会被插到左子树中去,所以当root等于none的时候,也就是这个节点应该放的位置, 这 里直接return treenode(val)就可以了。然后分别对左子树和右子树进行递归,这里是需要用root.lefta&root.right来接收左右孩子的值的。所以这里是一个典型的 递归函数需要变量接受返回值
450. 删除二叉搜索树中的节点
思路:这道题吧 面试如果考这个 我就直接g了。。。。
有以下五种情况:
- 第一种情况:没找到删除的节点,遍历到空节点直接返回了
- 找到删除的节点
卡哥给的动画 直接秒懂!!!(from 代码随香录)
- 第二种情况:左右孩子都为空(叶子节点),直接删除节点, 返回NULL为根节点
- 第三种情况:删除节点的左孩子为空,右孩子不为空,删除节点,右孩子补位,返回右孩子为根节点
- 第四种情况:删除节点的右孩子为空,左孩子不为空,删除节点,左孩子补位,返回左孩子为根节点
- 第五种情况:左右孩子节点都不为空,则将删除节点的左子树头结点(左孩子)放到删除节点的右子树的最左面节点的左孩子上,返回删除节点右孩子为新的根节点。
-
代码:
-
def deleteNode(self, root: Optional[TreeNode], key: int) -> Optional[TreeNode]: if not root: return None if root.val==key: #如果是叶子节点 if root.left==None and root.right==None: return None #左不空 右空 elif root.left!=None and root.right==None: return root.left #左空 右不空 elif root.left==None and root.right!=None: return root.right else: #左右都不空 选择让右孩子继位 cur = root.right while cur.left!=None: cur = cur.left cur.left = root.left return root.right if key < root.val: root.left = self.deleteNode(root.left,key) if key >root.val: root.right = self.deleteNode(root.right,key) return root
-