搜索二叉树中等 LeetCode701. 二叉搜索树中的插入操作
描述
给定二叉搜索树(BST)的根节点和要插入树中的值,将值插入二叉搜索树。 返回插入后二叉搜索树的根节点。 输入数据 保证 ,新值和原始二叉搜索树中的任意节点值都不同。
注意,可能存在多种有效的插入方式,只要树在插入后仍保持为二叉搜索树即可。 你可以返回 任意有效的结果 。
分析
遍历搜索二叉树,当遇到null时,说明找到了符合的位置,new一个新结点返回就好。 不要全遍历搜索二叉树,利用搜索二叉树的特性去遍历。也可以迭代实现,需要一个变量记录要插入结点的父节点。
class Solution { public TreeNode insertIntoBST(TreeNode root, int val) { if(root == null){ return new TreeNode(val); } if(val > root.val){ root.right = insertIntoBST(root.right,val); } if(val < root.val){ root.left = insertIntoBST(root.left,val); } return root; } }
下一篇:
牛客网 字节跳动算法题-特征提取