每日一题~二叉搜索树中的插入操作
题目链接:701. 二叉搜索树中的插入操作 - 力扣(LeetCode)
题目描述:
思路分析:由题可知,题目的要求是给我们一个二叉搜索树和一个 val,将这个 val 插入到二叉搜索树中,并且这个树仍然是二叉搜索树。题目中给了两个插入方式,第一种是将 val 插入到原来没有节点的位置;第二种是,将 val 替换掉已经存在的节点,然后重新调整树。通过观察发现第一种方法比第二种更加简单,所以我们接下来就根据第一种方法进行解答。
插入 val 节点总共需要两步:
1、寻找 val 节点插入的位置, 因为题目中的树是一个二叉搜索树,所以如果 root.val < val ,那么 val 应该插到右子树那边,我们只需要在右子树中再寻找合适的位置就可以了,如果 root.val > val, 那么从左子树中再寻找位置就可以。
2、插入 val 节点,如何确定这个位置是我们要找的位置呢?由第一步可以知道,我们是从 root 节点开始向下找,所以当 root = null 时,说明我们已经找到了那个位置,这时候我们直接返回到上一个节点,如果此时 root.left == null && root.val > val,那么直接将 val 添加到 root.left 就可以,注意右边的情况和左边判断时一样,不能直接使用 else,如果直接使用 else,那么在回溯到中间的时候会将 val 重新添加到其他节点的右子树。
代码示例:
class Solution {
public TreeNode insertIntoBST(TreeNode root, int val) {
if(root == null) return new TreeNode(val);
search(root,val);
return root;
}
public void search(TreeNode root,int val) {
if(root == null) return;
if(root.val > val) {
// 在左子树继续寻找
search(root.left,val);
}else {
search(root.right,val);
}
if(root.left == null && root.val > val) {
root.left = new TreeNode(val);
}else if(root.right == null && root.val < val){
root.right = new TreeNode(val);
}
}
}