二叉搜索树中插入一个数-leetcode-701 . 二叉搜索树中的插入操作

二叉搜索树中的插入操作

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

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

1、给定二叉搜索树:

        4
       / \
      2   7
     / \
    1   3

和 插入的值: 5

返回这个二叉搜索树:

         4
       /   \
      2     7
     / \   /
    1   3 5

 利用二叉搜索树的定义:

  1. 每个节点中的值必须大于(或等于)存储在其左侧子树中的任何值。
  2. 每个节点中的值必须小于(或等于)存储在其右子树中的任何值。

 递归实现

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

 从上到下跑一遍,与每个节点比较,

  public TreeNode insertIntoBST(TreeNode root, int val) {
        if(root == null)
        return new TreeNode(val);
        TreeNode tnode = root;
        while(tnode != null){
            if(tnode.val > val){
                if(tnode.left == null){
                    tnode.left = new TreeNode(val);
                    break;
                }
                else
                    tnode = tnode.left;
            }
            else{
                if(tnode.right == null){
                    tnode.right = new TreeNode(val);
                    break;
                }
                else
                    tnode = tnode.right;
            }
        }
        return root;
    }

猜你喜欢

转载自blog.csdn.net/xiaomingds/article/details/109082505