一刷59-二叉树-700二叉搜索树中的搜索(e)

题目:
给定二叉搜索树(BST)的根节点和一个值。 
你需要在BST中找到节点值等于给定值的节点。 返回以该节点为根的子树。 如果节点不存在,则返回 NULL。
-------------
示例:
给定二叉搜索树:

        4
       / \
      2   7
     / \
    1   3

和值: 2
你应该返回如下子树:

      2     
     / \   
    1   3
    在上述示例中,如果要找的值是 5,但因为没有节点值为 5,我们应该返回 NULL。
-------------------------
思考:
之前我们讲了都是普通二叉树,那么接下来看看二叉搜索树。

二叉搜索树是一个有序树:

若它的左子树不空,则左子树上所有结点的值均小于它的根结点的值;
若它的右子树不空,则右子树上所有结点的值均大于它的根结点的值;
它的左、右子树也分别为二叉搜索树
这就决定了,二叉搜索树,递归遍历和迭代遍历和普通二叉树都不一样。

本题,其实就是在二叉搜索树中搜索一个节点。那么我们来看看应该如何遍历

递归法:
确定递归函数的参数和返回值
递归函数的参数传入的就是根节点和要搜索的数值,返回的就是以这个搜索数值所在的节点。

代码如下:

TreeNode* searchBST(TreeNode* root, int val)

确定终止条件
如果root为空,或者找到这个数值了,就返回root节点。

if (root == NULL || root->val == val) return root;

确定单层递归的逻辑
看看二叉搜索树的单层递归逻辑有何不同。

因为二叉搜索树的节点是有序的,所以可以有方向的去搜索。

如果root->val > val,搜索左子树,如果root->val < val,就搜索右子树,
最后如果都没有搜索到,就返回NULL。

代码如下:

if (root->val > val) return searchBST(root->left, val); // 注意这里加了return
if (root->val < val) return searchBST(root->right, val);
return NULL;

这里可能会疑惑,在递归遍历的时候,什么时候直接return 递归函数的返回值,什么时候不用加这个 return呢。

因为搜索到目标节点了,就要立即return了,这样才是找到节点就返回(搜索某一条边),
如果不加return,就是遍历整棵树了。
整体代码如下:

class Solution {
public:
    TreeNode* searchBST(TreeNode* root, int val) {
        if (root == NULL || root->val == val) return root;
        if (root->val > val) return searchBST(root->left, val);
        if (root->val < val) return searchBST(root->right, val);
        return NULL;
    }
};
迭代法:
一提到二叉树遍历的迭代法,可能立刻想起使用栈来模拟深度遍历,使用队列来模拟广度遍历。

对于二叉搜索树可就不一样了,因为二叉搜索树的特殊性,也就是节点的有序性,
可以不使用辅助栈或者队列就可以写出迭代法。

对于一般二叉树,递归过程中还有回溯的过程,例如走一个左方向的分支走到头了,那么要调头,在走右分支。

而对于二叉搜索树,不需要回溯的过程,因为节点的有序性就帮我们确定了搜索的方向。

例如要搜索元素为3的节点,我们不需要搜索其他节点,也不需要做回溯,查找的路径已经规划好了。

中间节点如果大于3就向左走,如果小于3就向右走

class Solution {
public:
    TreeNode* searchBST(TreeNode* root, int val) {
        while (root != NULL) {
            if (root->val > val) root = root->left;
            else if (root->val < val) root = root->right;
            else return root;
        }
        return NULL;
    }
};
本篇我们介绍了二叉搜索树的遍历方式,因为二叉搜索树的有序性,遍历的时候要比普通二叉树简单很多。

但是一些同学很容易忽略二叉搜索树的特性,所以写出遍历的代码就未必真的简单了。

所以针对二叉搜索树的题目,一样要利用其特性。

文中我依然给出递归和迭代两种方式,可以看出写法都非常简单,就是利用了二叉搜索树有序的特点。
------------------------------
代码:
解法一:递归法:普通二叉树
class Solution{
	public TreeNode searchBST(TreeNode root,int val){
		//如果root为空,或者找到这个数值了,就返回root节点。
		if(root==null || root.val==val)return root;
		//递归左子树
		TreeNode left  = searchBST(root.left,val);
		if(left!=null) return left;
		return searchBST(root.right,val);
	}
}
解法二:   递归,利用二叉搜索树特点,优化
class Solution{
	public TreeNode searchBST(TreeNode root,int val){
		//如果root为空,或者找到这个数值了,就返回root节点。
		if(root==null || root.val==val)return root;
		if(val<root.val){
			return searchBST(root.left,val);
		}else{
			return searchBST(root.right,val);
		}
	}
}
解法三:迭代,普通二叉树 使用栈 推荐!!!
class Solution{
	public TreeNode searchBST(TreeNode root,int val){
		if(root==null || root.val==val)return root;//特判
		Stack<TreeNode> stack = new Stack<>();//定义栈
		stack.push(root);//入栈
		while(!stack.isEmpty()){
			TreeNode node = stack.pop();//出栈
			if(node.val==val)return node;
			if(node.right!=null) stack.push(node.right);
			if(node.left!=null) stack.push(node.left);
		}
		return null;
	}
}
解法四:迭代,利用二叉搜索树特点,优化,可以不需要栈
class Solution{
	public TreeNode searchBST(TreeNode root,int val){
		while(root!=null){
			if(val>root.val) root = root.right;
			else if(val<root.val) root = root.left;
			else return root;
		}
		return root;
	}
}

力扣链接

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值