JavaLeetCode刷题记录算法提高之LeetCode刷题

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

2019-08-21  本文已影响3人  TheKey_

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

给定二叉搜索树(BST)的根节点和要插入树中的值,将值插入二叉搜索树。 返回插入后二叉搜索树的根节点。 保证原始二叉搜索树中不存在新值。

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

示例:
给定二叉搜索树:

        4
       / \
      2   7
     / \
    1   3

和 插入的值: 5
你可以返回这个二叉搜索树:

         4
       /   \
      2     7
     / \   /
    1   3 5
或者这个树也是有效的:

         5
       /   \
      2     7
     / \   
    1   3
         \
          4

来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/insert-into-a-binary-search-tree/
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。


public class TreeNode {
        int val;
        TreeNode left;
        TreeNode right;

        TreeNode(int x) {
            val = x;
        }
    }

思路:

  1. 递归终止条件为当前节点为空,创建一个新的节点即可
  2. 判断要插入的值 val 和根节点的大小,如果 val < root.val,则插入左子树
  3. 如果 val > root.val,则插入右子树
public TreeNode insertIntoBST(TreeNode root, int val) {
        if (root == null) return new TreeNode(val);
        if (val < root.val) root.left =  insertIntoBST(root.left, val);
        else if (val > root.val) root.right =  insertIntoBST(root.right, val);
        return root;
    }

复杂度分析:


上一篇 下一篇

猜你喜欢

热点阅读