[代码随想录]Day19-二叉树part08

合集 - 代码随想录(52)

26.[代码随想录]Day19-二叉树part082023-08-16

27.[代码随想录]Day18-二叉树part072023-08-15 28.[代码随想录]Day17-二叉树part062023-08-14 29.[代码随想录]Day16-二叉树part052023-08-12 30.[代码随想录]Day15-二叉树part042023-08-11 31.[代码随想录]Day52-单调栈part032023-09-23 32.[代码随想录]Day51-单调栈part022023-09-22 33.[代码随想录]Day50-单调栈part012023-09-21 34.[代码随想录]Day49-动态规划part172023-09-20 35.[代码随想录]Day48-动态规划part162023-09-19 36.[代码随想录]Day47-动态规划part152023-09-18 37.[代码随想录]Day46-动态规划part142023-09-16 38.[代码随想录]Day45-动态规划part132023-09-15 39.[代码随想录]Day44-动态规划part122023-09-14 40.[代码随想录]Day43-动态规划part112023-09-13 41.[代码随想录]Day42-动态规划part102023-09-12 42.[代码随想录]Day41-动态规划part092023-09-11 43.[代码随想录]Day40-动态规划part082023-09-09 44.[代码随想录]Day39-动态规划part072023-09-08 45.[代码随想录]Day38-动态规划part062023-09-07 46.[代码随想录]Day37-动态规划part052023-09-06 47.[代码随想录]Day36-动态规划part042023-09-05 48.[代码随想录]Day35-动态规划part032023-09-04 49.[代码随想录]Day34-动态规划part022023-09-02 50.[代码随想录]Day33-动态规划part012023-09-01 51.[代码随想录]Day32-贪心算法part062023-08-31 52.[代码随想录]Day31-贪心算法part052023-08-30

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

思路：

BST和普通二叉树不同的一点是可以根据特性来找最近公共祖先，只要找到第一个值比p大比q小(假设p<q)的节点返回即可。

代码：

/**
 * Definition for a binary tree node.
 * type TreeNode struct {
 *     Val   int
 *     Left  *TreeNode
 *     Right *TreeNode
 * }
 */

func lowestCommonAncestor(root, p, q *TreeNode) *TreeNode {
    if root == nil {
        return nil
    }
    if p.Val > q.Val {
        p, q = q, p
    } // 保证p < q 范围取(p,q)
    return step(root,p,q)
}
func step(root, p, q *TreeNode) *TreeNode{
    if root == nil {
        return nil
    }
    if root.Val > q.Val { // 大于q，向左找
        left := step(root.Left, p, q)
        if left != nil {
            return left
        }
    }
    if root.Val < p.Val { // 小于p，向右找
        right := step(root.Right, p, q)
        if right != nil {
            return right
        }
    }
    return root //  现在是大于p 小于q，返回
}

参考：

代码随想录

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

思路：

BST的插入只需要找到第一个：

比val大且没有左子树的位置
比val小且没有右子树的位置

之后在该位置插入节点，向上递归返回即可。

代码：

/**
 * Definition for a binary tree node.
 * type TreeNode struct {
 *     Val int
 *     Left *TreeNode
 *     Right *TreeNode
 * }
 */
func insertIntoBST(root *TreeNode, val int) *TreeNode {
    if root == nil { // 如果是空插入
        return &TreeNode{Val:val,}
    }
    if val < root.Val { // 向左找，找到后返回
        root.Left = insertIntoBST(root.Left, val)
        return root
    }
    if val > root.Val { // 向右找，找到后返回
        root.Right = insertIntoBST(root.Right, val)
        return root
    }
    return nil
}

参考：

代码随想录

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

思路：

和上一题类似，这一题要先找到值为key的节点位置，然后把该节点的左子树放到右子树的最左下角。

代码：

/**
 * Definition for a binary tree node.
 * type TreeNode struct {
 *     Val int
 *     Left *TreeNode
 *     Right *TreeNode
 * }
 */
func deleteNode(root *TreeNode, key int) *TreeNode {
    if root == nil {
        return nil
    }
    if key < root.Val {
        root.Left = deleteNode(root.Left, key)
        return root
    }
    if key > root.Val {
        root.Right = deleteNode(root.Right, key)
        return root
    }
    // 下面是key == root.Val的情况
    if root.Right == nil {
        return root.Left
    }
    if root.Left == nil {
        return root.Right
    }
    left := root.Left // 记录删除节点左子树
    right := root.Right // 记录删除节点右子树
    root = root.Right // 进入右子树
    for root.Left != nil { // 找到最左侧节点
        root = root.Left
    }
    root.Left = left // 把左子树放到这里
    return right // 返回记录的右子树
}