701.二叉搜索树中的插入操作
给定二叉搜索树(BST)的根节点 root 和要插入树中的值 value ,将值插入二叉搜索树。 返回插入后二叉搜索树的根节点。 输入数据 保证 ,新值和原始二叉搜索树中的任意节点值都不同。
注意,可能存在多种有效的插入方式,只要树在插入后仍保持为二叉搜索树即可。 你可以返回 任意有效的结果 。
输入:root = [4,2,7,1,3], val = 5 输出:[4,2,7,1,3,5] 解释:另一个满足题目要求可以通过的树是:
根据二叉搜索树的性质:
node 的左子树结点值一定小于 node 的值
node 的右子树结点值一定大于 node 的值
node 的左右子树均为二叉搜索树
我们可以使用递归的思想:
1. 如果根节点的值 小于 val,那么就将根节点的右子树节点作为根节点,继续进行递归
2. 如果根节点的值 大于 val,那么就将根节点的左子树节点作为根节点,继续进行递归
如果当出现节点为空时,那么就可以将val插入,返回一个新结点。
class Solution {
public TreeNode insertIntoBST(TreeNode root, int val) {
if(root == null){ // root == null 为终止条件,此时新建结点,并返回插入
TreeNode node = new TreeNode(val);
return node;
}
if(root.val > val){ // val 比root值小,对左子树结点进行递归操作
root.left = insertIntoBST(root.left, val);
}
if(root.val < val){ // val 比root值大,对右子树结点进行递归操作
root.right = insertIntoBST(root.right, val);
}
return root;
}
}