29_二叉搜索树中的插入操作

合集 - 数据结构与算法(61)

1.01_设计一个有getMin功能的栈2023-09-222.02_由两个栈组成的队列2023-09-263.03_如何仅用递归函数和栈操作逆序一个栈2023-09-274.04_猫狗队列2023-10-055.05_用一个栈实现另一个栈的排序2023-10-066.06_用栈来求解汉诺塔问题2023-10-087.07_用队列实现栈2023-10-098.09_删除字符串中的所有相邻重复项2023-10-139.08_ 有效的括号2023-10-1110.10_逆波兰表达式求值2023-10-1611.11_滑动窗口最大值2023-10-1712.12_前K个高频元素2023-10-1813.01_移除链表元素2023-10-2314.02_设计链表2023-10-2415.03_反转链表2023-10-2516.04_两两交换链表中的节点2023-10-2617.05_删除链表的倒数第N个节点2023-10-2718.06_链表相交2023-10-3119.07_环形链表2023-11-0420.01_二叉树的递归遍历2023-11-0621.二叉树理论基础2023-11-0622.02_二叉树的迭代遍历2023-11-1023.04_二叉树的层序遍历2023-11-1024.05_二叉树的层次遍历II2023-11-1625.06_二叉树的右视图2023-11-1726.07_二叉树的层平均值2023-11-2027.08_N叉树的层序遍历2023-11-2328.09_每个行中找最大值2023-11-2329.10_填充每个节点的下一个右侧节点指针2023-11-2430.11_二叉树的最大深度2023-11-2531.12_二叉树的最小深度2023-11-2732.13_翻转二叉树2023-11-2833.14_对称二叉树2023-11-3034.15_完全二叉树的节点个数2023-12-0635.16_平衡二叉树2023-12-0636.17_二叉树的所有路径2023-12-0637.18_左叶子之和2023-12-0738.19_找树左下角的值2023-12-0839.20_路径总和2023-12-0940.21_从中序与后序遍历序列构造二叉树2023-12-2041.22_最大二叉树2023-12-2142.23_合并二叉树2023-12-2243.24_二叉搜索树中的搜索2023-12-2544.27_二叉搜索树的众数2024-01-0945.28_二叉树的最近公共祖先2024-01-09

46.29_二叉搜索树中的插入操作2024-01-18

47.30_删除二叉搜索树中的节点2024-01-2048.31_修剪二叉搜索树2024-01-2049.32_将有序数组转换为平衡二叉搜索树2024-01-2350.33_把二叉搜索树转换为累加树2024-01-2451.动态规划理论2024-05-2452.01_斐波那契数列2024-05-2453.02_爬楼梯2024-05-2454.03_使用最小花费爬楼梯2024-05-2455.04_不同路径2024-05-2556.05_不同路径2(带障碍物版)2024-05-2557.06_整数拆分2024-05-2758.08_杨辉三角2024-05-3059.10_最后一块石头的重量2024-06-0560.09_分割等和子集2024-06-0561.74_搜索二维矩阵2024-07-20

收起

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

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

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

示例 1：

输入：root = [4,2,7,1,3], val = 5
输出：[4,2,7,1,3,5]
解释：另一个满足题目要求可以通过的树是：

示例 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]

【思路】

递归

递归三部曲：

• 确定递归函数参数以及返回值

  TreeNode insertIntoBST(TreeNode root, int val);
  

• 确定终止条件

  终止条件就是找到遍历的节点为null的时候，就是要插入节点的位置了，并把插入的节点返回，代码如下：

  if (root == null) 
  	return new TreeNode(val);
  

• 确定单层递归的逻辑

  if (root.val > val) {
  	root.left = insertInToBST(root.left, val);
  }
  if (root.val < val) {
  	root.right = insertIntoBST(root.right, 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;
      }
  }
  

  递归法

  class Solution {
      public TreeNode insertIntoBST(TreeNode root, int val) {
          if (root == null) // 如果当前节点为空，也就意味着val找到了合适的位置，此时创建节点直接返回。
              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;
      }
  }