题目描述:
给定二叉搜索树(BST)的根节点 root 和一个整数值 val。
你需要在 BST 中找到节点值等于 val 的节点。 返回以该节点为根的子树。 如果节点不存在,则返回 null 。
来源:力扣(LeetCode)
思路:
了解BST的性质,左叶子节点小于根节点,右叶子节点大于根节点
迭代:借助于自己声明的 数据结构,实现迭代!
栈结构:可以实现二叉树深度优先搜索的迭代;
队列:可以实现二叉树广度优先的迭代!
代码:
1)递归:利用BST性质,分情况递归
class Solution {
public TreeNode searchBST(TreeNode root, int val) {
//二叉搜索树BST:特点是中序遍历,大小是升序的! 平衡二叉搜索树是加了高度 AVL
if(root == null) return null;
if(root.val == val) return root;
else if(root.val < val) return searchBST(root.right,val);
else return searchBST(root.left,val);
}
}
2)迭代
class Solution {
public TreeNode searchBST(TreeNode root, int val) {
//迭代
while(root != null){
if(root.val == val) return root;
else root = root.val < val ? root.right : root.left;
}
return root;
}
}