https://leetcode-cn.com/problems/insert-into-a-binary-search-tree/
题意
给定二叉搜索树(BST)的根节点和要插入树中的值,将值插入二叉搜索树。 返回插入后二叉搜索树的根节点。 输入数据 保证 ,新值和原始二叉搜索树中的任意节点值都不同。
注意,可能存在多种有效的插入方式,只要树在插入后仍保持为二叉搜索树即可。 你可以返回 任意有效的结果 。
输入:root = [4,2,7,1,3], val = 5
输出:[4,2,7,1,3,5]
解释:另一个满足题目要求可以通过的树是:
题解
往二叉搜索树中插入元素,且插入的元素与树中的元素不相同。
迭代解法使用常规思路,找到插入的位置,创建新节点与父节点相连,只不过是作为左节点还是右节点需要另外判断。
迭代解法
/**
* 迭代解法
*/
class Solution {
public TreeNode insertIntoBST(TreeNode root, int val) {
if(root==null)//树为空
return new TreeNode(val);
TreeNode node=root;
while(true){
if(val<node.val){
if(node.left!=null){//若还有左子节点,则继续迭代
node=node.left;
}else {//否则就应放在当前节点左侧
node.left = new TreeNode(val);
return root;
}
}else {
if(node.right!=null) {//若还有右子节点,则继续迭代
node = node.right;
}else {//否则就应放在当前节点右侧
node.right = new TreeNode(val);
return root;
}
}
}
}
}
递归解法
递归解法,先明确insertIntoBST方法的功能,该方法应该在根节点为root的树中插入元素为val的新节点,并返回插入完成后的头节点。假如对root的左节点使用该方法,则表示在根节点为root.left的树中插入元素为val的新节点,并返回插入完成后的头节点。此时,左子树的头节点可能因插入操作而产生变化,在此需要更新root的左节点(比如左子树本来为空,插入多了一个节点),右子树也是相同的操作。
root.left=insertIntoBST(root.left,val);
root.right=insertIntoBST(root.right,val);
所以只需要找到新节点插入的大致方向(左或者右),然后交给左子树或右子树去处理即可。
再来考虑边界条件:什么找到真正该插入的位置?将插入操作不断的交给左、右子树去处理,当左或右节点为空时,就是应该插入的位置。此时只需要创建一个新节点返回,交给父节点跟新即可。
class Solution {
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在右子树中插入并跟新左节点
root.right=insertIntoBST(root.right,val);
return root;
}
}