二叉搜索树中的插入操作｜Go主题月

深海001
1,535 阅读2分钟

【Golang主题学习月】 刷题比玩游戏好多了，成就感越来越强，每天坚持刷一道题，每天锻炼30分钟，等8块腹肌,等大厂offer.

😄

 \color{red}{~}

我相信，如果在面试中遇到此题，逻辑清晰、正确表达出来、手撕

应该会超过一部分的面试者。

对树不熟悉的朋友，可以看看前面的基础训练题哦！

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

给定二叉搜索树（BST）的根节点和要插入树中的值，将值插入二叉搜索树。 返回插入后二叉搜索树的根节点。 输入数据 保证 ，新值和原始二叉搜索树中的任意节点值都不同。

注意，可能存在多种有效的插入方式，只要树在插入后仍保持为二叉搜索树即可。 你可以返回 任意有效的结果 。

图片.png

示例 2：

输入：root = [40,20,60,10,30,50,70], val = 25

输出：[40,20,60,10,30,50,70,null,null,25]

示例 3：

输入：root = [4,2,7,1,3,null,null,null,null,null,null], val = 5

输出：[4,2,7,1,3,5]

参考代码

定义一颗树

/**
 * Definition for a binary tree node.
 * type TreeNode struct {
 *     Val int          // 根
 *     Left *TreeNode   //左节点
 *     Right *TreeNode  //右节点
 * }
 */

二叉搜索树，左边比根节点小，右边比根节点大

GO语言版 迭代

  1. 根据根节点区分左子树或者右子树。

  2. 如果在左子树，找到判断左子树是否为空，为空，建立一颗树，跳出循环。

  3. 如果在右子树，注意有可能时左子树的右子树

func insertIntoBST(root *TreeNode, val int) *TreeNode {
    if root == nil {
        return &TreeNode{Val: val}
    }
    p := root
    for p != nil {
        if val < p.Val {  // 1.
            if p.Left == nil {  // 2.
                p.Left = &TreeNode{Val: val}
                break
            }
            p = p.Left
        } else {   // 3.
            if p.Right == nil {
                p.Right = &TreeNode{Val: val}
                break
            }
            p = p.Right
        }
    }
    return root
}

java

class Solution {
    public TreeNode insertIntoBST(TreeNode root, int val) {
        if (root == null) {
            return new TreeNode(val);
        }
        TreeNode pos = root;
        while (pos != null) {
            if (val < pos.val) {   // 是否小于根节点
                if (pos.left == null) {  // 左节点为空，目标值就赋值为左节点
                    pos.left = new TreeNode(val);
                    break;
                } else {    // 继续找左节点
                    pos = pos.left;
                }
            } else {      // 大于根节点或左节点
                if (pos.right == null) {
                    pos.right = new TreeNode(val);  
                    break;
                } else {
                    pos = pos.right;
                }
            }
        }
        return root;
    }
}


// 递归

public static TreeNode insertIntoBST(TreeNode root, int val) {
        if(root == null) {
            return new TreeNode(val);
        }
        // 大右小左
        if(root.val < val) {
            root.right = insertIntoBST(root.right, val);
        }else if(root.val > val) {
            root.left = insertIntoBST(root.left, val);
        }
        return root;
    }

真心感谢帅逼靓女们能看到这里，如果这个文章写得还不错，觉得有点东西的话

求点赞👍 求关注❤️ 求分享👥 对8块腹肌的我来说真的 非常有用！！！

如果本篇博客有任何错误，请批评指教，不胜感激 ！❤️❤️❤️❤️

avatar
文章
阅读
粉丝