代码随想录之二叉搜索树的插入操作

最新推荐文章于 2024-07-05 16:42:37 发布

念念大侠

最新推荐文章于 2024-07-05 16:42:37 发布

阅读量433

点赞数

分类专栏：算法文章标签： leetcode 算法职场和发展

本文链接：https://blog.csdn.net/Derrick_itRose/article/details/131706020

版权

算法专栏收录该内容

41 篇文章 0 订阅

订阅专栏

本题在力扣701

本体思路学习于代码随想录

JAVA版本

思路：本题的思想较为简单，直接找到适合插入的结点来进行插入即可

解法一：
递归法：

class Solution {
    public TreeNode insertIntoBST(TreeNode root, int val) {
        if(root ==null) {
            return new TreeNode(val);
        }
        if(root.val < val){
            root.right = insertIntoBST(root.right,val);
        } else {
            root.left = insertIntoBST(root.left,val);
        }
        return root;
    }
}

解法二：
迭代法：
迭代法的思想就是找到最适合插入的结点，在最适合插入的结点来进行插入操作。

class Solution {
    public TreeNode insertIntoBST(TreeNode root, int val) {
         if (root == null) return new TreeNode(val);

        TreeNode newRoot = root;
        TreeNode pre = root;
        while(root != null) {
            pre = root;
            if (root.val > val) {
                root = root.left;
            } else if (root.val < val) {
                root = root.right;
            } 
        }
        if (pre.val > val) {
            pre.left = new TreeNode(val);
        } else {
            pre.right = new TreeNode(val);
        }

        return newRoot;
    }
}