LeetCode700. 二叉搜索树中的搜索（Java）

最新推荐文章于 2023-04-29 19:09:03 发布

原创最新推荐文章于 2023-04-29 19:09:03 发布 · 229 阅读

0 ·

CC 4.0 BY-SA版权

LeetCode刷题专栏收录该内容

52 篇文章

订阅专栏

本文介绍了两种在二叉搜索树(BST)中查找特定值的节点的方法：递归和迭代。通过示例代码详细解释了如何实现这些算法，并讨论了在节点不存在时的返回值。

题目：

给定二叉搜索树（BST）的根节点和一个值。你需要在BST中找到节点值等于给定值的节点。返回以该节点为根的子树。如果节点不存在，则返回 NULL。

示例：
在这里插入图片描述
代码：

解法一

/**
 * Definition for a binary tree node.
 * public class TreeNode {
 *     int val;
 *     TreeNode left;
 *     TreeNode right;
 *     TreeNode(int x) { val = x; }
 * }
 */
 //递归的思想
class Solution {
    public TreeNode searchBST(TreeNode root, int val) {
        if(root==null){	//如果根节点为空
           return null;
        }
        if(root.val>val){   //要找的值大于当前元素 向其左子树寻找
           return searchBST(root.left,val);
        }
        if(root.val<val){   //要找的值大于当前元素 向其右子树寻找
           return searchBST(root.right,val);
        }
        return root;    //找到元素
    }
}

别人的代码

/**
 * Definition for a binary tree node.
 * public class TreeNode {
 *     int val;
 *     TreeNode left;
 *     TreeNode right;
 *     TreeNode(int x) { val = x; }
 * }
 */
 //迭代的思想
class Solution {
    public TreeNode searchBST(TreeNode root, int val) {
        if(root == null) {
            return null;
        }
        while(root != null) {
            if(val < root.val) {//要找的值大于当前元素 向其左子树寻找
                root = root.left;
            }else if(val > root.val) {  //要找的值大于当前元素 向其右子树寻找
                root = root.right;
            }else {
                return root;
            }
        }
        return null;
    }
}