题目
给定二叉搜索树(BST)的根节点和要插入树中的值,将值插入二叉搜索树。 返回插入后二叉搜索树的根节点。
输入数据保证,新值和原始二叉搜索树中的任意节点值都不同。注意,可能存在多种有效的插入方式,只要树在插入后仍保持为二叉搜索树即可。 你可以返回任意有效的结果。
返回:
或者
思路
其实可以不考虑题目中提示所说的改变树的结构的插入方式,二叉搜索树的插入不用考虑重构问题。
只需要按照二叉树搜索树的规则去遍历,然后遇到空节点就直接插入就可以了。
● 递归法(三部曲):
- 递归函数参数和返回值:参数为根节点和待插入元素值,返回值为节点类型
- 终止条件:找到空节点即遍历到null的位置,就是要插入节点的位置了,并把插入的节点返回(返回是为了方便父子关系赋值,如将返回的值用root.left接住即可完成父子节点关系赋值)
- 确定单层递归的逻辑:不需要遍历整棵树,因为这是搜索树,遍历整棵搜索树是对搜索树的侮辱(doge),搜索树是有方向的,可以根据插入元素的数值,决定递归方向。
class Solution {
public TreeNode insertIntoBST(TreeNode root , int val){
if(root == null)
return new TreeNode(val);//如果当前节点为空,也就意味着val找到了合适的位置,此时创建节点直接返回。
if(root.val < val){//这里通过接住递归函数返回值,来完成父子节点的赋值操作,如root.right表示让root的左子树接住递归的返回值insertIntoBST(root.right,val),从而完成插入操作即让父节点指向孩子的操作
root.right = insertIntoBST(root.right,val);//递归创建右子树
}
if(root.val > val){
root.left = insertIntoBST(root.left,val);//递归创建左子树
}
return root;
}
}
● 迭代法:
在迭代法遍历的过程中,需要记录一下当前遍历节点的父节点,这样才能做插入节点的操作(即让root指向孩子)
class Solution {
public TreeNode insertIntoBST(TreeNode root, int val){
if(root == null) return new TreeNode(val);//如果当前节点为空,也就意味着val找到了合适的位置,此时创建节点直接返回。
TreeNode newRoot = root;//方便操作
TreeNode pre =root;
while(root != null){
pre = root;//从根节点开始搜索BST,根据有序性找到插入的位置的前驱位置,并且让pre指向这个位置,放在if语句前面是为了可以指向前驱位置
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;
}
}