题目:
给定二叉搜索树(BST)的根节点
root
和一个整数值val
。你需要在 BST 中找到节点值等于
val
的节点。 返回以该节点为根的子树。 如果节点不存在,则返回null
。输入:root = [4,2,7,1,3], val = 2 输出:[2,1,3]
二叉搜索树的特点:
① 根节点左子树的所有节点值都小于根节点
② 根节点右子树的所有节点值都大于根节点
③ 所有的子树也是二叉搜索树
根据二叉搜索树的性质,我们可以有递归法和迭代发两种解法:
递归法:
class Solution {
public TreeNode searchBST(TreeNode root, int val) {
if(root == null){
return null;
}
if(root.val == val){
return root;
}
return val < root.val ? searchBST(root.left,val) : searchBST(root.right,val);
}
}
迭代法:
自己第一次利用队列解法:
class Solution {
public TreeNode searchBST(TreeNode root, int val) {
if(root == null){
return null;
}
Queue<TreeNode> queue = new LinkedList<TreeNode>();
queue.offer(root);
while(!queue.isEmpty()){
TreeNode node = queue.poll();
if(node.val == val){
return node;
}else{
if(node.left != null && val < node.val){
queue.offer(node.left);
continue;
}
if(node.right != null && val > node.val){
queue.offer(node.right);
}
}
}
return null;
}
}
查看官方题解之后解法:
class Solution {
public TreeNode searchBST(TreeNode root, int val) {
if(root == null){
return null;
}
while(root != null){
if(root.val == val){
return root;
}
//二叉搜索树的特点
root = val < root.val ? root.left : root.right;
}
return null;
}
}