代码随想录算法训练营第二十二天

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

题目链接：二叉搜索树的最近公共祖先
可以利用搜索二叉树的特性，如果p，q的值都比root小就直接去左边，如果都大就直接去右边，如果在两边的就直接输出root。

class Solution:
    def lowestCommonAncestor(self, root: 'TreeNode', p: 'TreeNode', q: 'TreeNode') -> 'TreeNode':
        if not root:
            return root
        if (p == root) or (q == root):
            return root
        if root.val > max(p.val,q.val):
            return self.lowestCommonAncestor(root.left,p,q)
        elif root.val < min(p.val,q.val):
            return self.lowestCommonAncestor(root.right,p,q)
        else:
            return root

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

题目链接：二叉搜索树中的插入操作

递归

不知道怎么把node插到半当中，所以最后还是决定放到最后，所以就是val大于root就往右，小于root就往左，走到None就插入，所以除非是root本来就是None的情况，就都可以插入，如果root自己就是None的话，试过了直接在traversal里改但好像没什么用，所以还是在insertIntoBST里做一下条件判断了。

class Solution:
    def insertIntoBST(self, root: Optional[TreeNode], val: int) -> Optional[TreeNode]:
        def traversal(root,val):
            if not root:
                return
            if root.val > val:
                if not root.left:
                    root.left = TreeNode(val=val)
                    return 
                else:
                    traversal(root.left,val)
            elif root.val < val:
                if not root.right:
                    root.right = TreeNode(val=val)
                    return
                else:
                    traversal(root.right,val)
        traversal(root,val)
        if not root:
            return TreeNode(val=val)
        return root        

递归简易版

class Solution:
    def insertIntoBST(self, root: TreeNode, val: int) -> TreeNode:
        if not root: 
        	return TreeNode(val)
        if val < root.val: 
            root.left = self.insertIntoBST(root.left, val)
        if root.val < val:
            root.right = self.insertIntoBST(root.right, val)
        return root

没想到可以直接接上去。。。迭代比较简单，之后有空随便写一些把。

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

题目链接：删除二叉搜索树中的节点
血泪教训：如果在里面加了一个内函数，记得改名。
这里是几种情况：

• 找不到key，原样出来
• 找到key了：1. key的node左边有右边没有，直接接到上一个；2. 右边有左边没有也一样；3. 都没有的就直接不要了；4. 都有的情况下，先把左边接上去，然后把右孩子接到左孩子右边的最下面，因为左孩子所有元素都大于右孩子，所以直接接上就行。

class Solution:
    def deleteNode(self, root: Optional[TreeNode], key: int) -> Optional[TreeNode]:
        if not root:
            return root
        if root.val > key:
            #print(self.deleteNode(root.left, key))
            root.left = self.deleteNode(root.left, key)
        elif root.val < key:
            root.right = self.deleteNode(root.right, key)
        elif root.val == key:
            if (root.left) and (not root.right):
                return root.left
            elif (not root.left) and (root.right):
                return root.right
            elif root.right and root.right:
                left = root.left
                right = root.right
                cur = right
                while cur.left:
                    cur = cur.left
                cur.left = left
                return right
            else:
                return None
        return root          

顺便尝试了一下，写了内函数traversal，并没有成功，这种修改二叉树的以后还是别写traversal了，确实不需要两个。