给定二叉搜索树(BST)的根节点 root
和一个整数值 val
。
你需要在 BST 中找到节点值等于 val
的节点。 返回以该节点为根的子树。 如果节点不存在,则返回 null
。
示例 1:
输入:root = [4,2,7,1,3], val = 2 输出:[2,1,3]
Example 2:
输入:root = [4,2,7,1,3], val = 5 输出:[]
提示:
- 数中节点数在
[1, 5000]
范围内 1 <= Node.val <= 107
root
是二叉搜索树1 <= val <= 107
Related Topics
根据二叉搜索树的左小右大的特性【递归&遍历树】
递归
class Solution {
public TreeNode searchBST(TreeNode root, int val) {
if (root == null){
return null;
}
if (root.val == val){
return root;
}
return root.val > val ? searchBST(root.left, val) : searchBST(root.right, val);
}
}
应该没啥特殊的,二叉搜索树的特性
左子节点 < 根节点 < 右子节点
所以照着这个写就行了
也可以改递归为遍历的写法,毕竟 递归和遍历循环是可以相互转化的
遍历
class Solution {
public TreeNode searchBST(TreeNode root, int val) {
while (root !=null){
if (root.val == val) return root;
root = root.val > val? root.left : root.right;
}
return root;
}
}
发表评论