题目描述
题干:
给定二叉搜索树(BST)的根节点和一个值。 你需要在BST中找到节点值等于给定值的节点。
返回以该节点为根的子树。 如果节点不存在,则返回 NULL。
例如,给定二叉搜索树:
4
/ \
2 7
/ \
1 3
和值: 2
你应该返回如下子树:
2
/ \
1 3
在上述示例中,如果要找的值是 5,但因为没有节点值为 5,我们应该返回 NULL。
题解思路
返回以指定节点的二叉树,当然我们只需要返回该节点即可,根据二叉搜索树的性质
节点的左孩子永远比它小,右孩子节点永远比它大,所以我们可以递归寻找目标节点
正确代码
// 递归
public TreeNode searchBST(TreeNode root, int val) {
if (root == null) {
return null;
}
if (val == root.val) {
return root;
}
return searchBST(val < root.val ? root.left : root.right, val);
}
// 迭代
public TreeNode searchBST01(TreeNode root, int val) {
while (root != null) {
if (val == root.val) {
return root;
}
root = val < root.val ? root.left : root.right;
}
return null;
}
总结
就像剪枝思想一样,递归写法总是会产生多余的计算消耗,在斐波那契数组的经典问题中最高效的解法就是记录每次计算结果
如果文章存在问题或者有更好的题解,欢迎在评论区斧正和评论,各自努力,最高处见