代码随香录day21

235. 二叉搜索树的最近公共祖先

代码随香录day21_第1张图片

 代码随香录day21_第2张图片

 本题思路:

                还是要利用二叉搜索树的特性,中序遍历为有序数组。如果pq两个节点都小于root,那么最近公共祖肯定是在他的左子树,如果都大于那么,肯定就在右子树。然后直接return root 

代码:

        

def lowestCommonAncestor(self, root: 'TreeNode', p: 'TreeNode', q: 'TreeNode') -> 'TreeNode':
        if p.valroot.val and q.val>root.val:
            return self.lowestCommonAncestor(root.right,p,q)
        return root

701. 二叉搜索树中的插入操作

代码随香录day21_第3张图片

 代码随香录day21_第4张图片

本题目涉及到一个二叉搜索树树的插入操作,如果当root等于none的时候,实际上我们这里就要开始插入操作了,因为我们对二叉搜索树的特性有了解,如果这个val<根节点的话,它一定会被插到左子树中去,所以当root等于none的时候,也就是这个节点应该放的位置, 这 里直接return treenode(val)就可以了。然后分别对左子树和右子树进行递归,这里是需要用root.lefta&root.right来接收左右孩子的值的。所以这里是一个典型的 递归函数需要变量接受返回值

450. 删除二叉搜索树中的节点

代码随香录day21_第5张图片

代码随香录day21_第6张图片

 

 思路:这道题吧 面试如果考这个 我就直接g了。。。。

有以下五种情况:

  • 第一种情况:没找到删除的节点,遍历到空节点直接返回了
  • 找到删除的节点

     卡哥给的动画 直接秒懂!!!(from 代码随香录)

    • 第二种情况:左右孩子都为空(叶子节点),直接删除节点, 返回NULL为根节点
    • 第三种情况:删除节点的左孩子为空,右孩子不为空,删除节点,右孩子补位,返回右孩子为根节点
    • 第四种情况:删除节点的右孩子为空,左孩子不为空,删除节点,左孩子补位,返回左孩子为根节点
    • 第五种情况:左右孩子节点都不为空,则将删除节点的左子树头结点(左孩子)放到删除节点的右子树的最左面节点的左孩子上,返回删除节点右孩子为新的根节点。
    • 代码随香录day21_第7张图片
  • 代码:

    • 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

你可能感兴趣的:(算法,数据结构)