题目:
给定二叉搜索树(BST)的根节点 root
和一个整数值 val
。
你需要在 BST 中找到节点值等于 val
的节点。 返回以该节点为根的子树。 如果节点不存在,则返回 null
。
思路:递归法。
第一步:确定参数与返回值。参数为树节点root,整数值val,返回值为树节点
第二步:确定终止条件。如果root=null或者root.val=val时返回root
第三步:确定单层递归逻辑。根据二叉搜索树的性质,比根节点小的节点在根节点的左子树,比根节点大的节点在根节点的右子树。若根节点的值>val,递归遍历左子树;若根节点的值<val,递归右子树。
代码:
public TreeNode searchBST(TreeNode root, int val) {
if(root==null||root.val==val) return root;
if(root.val>val) return searchBST(root.left,val);
else return searchBST(root.right,val);
}