代码随香录day21

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