看见BST就要想到,中序遍历是有序的(升序)
文章目录
700.BST中的搜索
思路:
利用 BST 左小右大的特性,可以避免搜索整棵二叉树去寻找元素,从而提升效率。
代码实现:
class Solution {
public TreeNode searchBST(TreeNode root, int val) {
if(root == null){return null;}
//当root.val > val 就去左子树搜
if(root.val > val){
return searchBST(root.left,val);
}
//当root.val < val 就去右子树搜
if(root.val < val){
return searchBST(root.right,val);
}
return root;
}
}
701.BST中的插入
思路:
如果要递归地插入或者删除二叉树节点,递归函数一定要有返回值,而且返回值要被正确的接收。
插入的过程可以分两部分:
1、寻找正确的插入位置
2、把元素插进去,这就要把新节点以返回值的方式接到父节点上。
代码实现:
class Solution {
public TreeNode insertIntoBST(TreeNode root, int val) {
// 找到空位置插入新节点
if (root == null) return new TreeNode(val);
// if (root.val == val)
// BST 中一般不会插入已存在元素
if (root.val < val)
root.right = insertIntoBST(root.right, val);
if (root.val > val)
root.left = insertIntoBST(root.left, val);
return root;
}
}