LeetCode算法题-Search in a Binary Search Tree(Java实现)

这是悦乐书的第295次更新,第314篇原创

01 看题和准备

今天介绍的是LeetCode算法题中Easy级别的第163题(顺位题号是700)。给定一个二叉搜索树(BST)的和正整数val。 你需要在BST中找到节点的值等于给定val的节点。返回以该节点为根的子树。如果此节点不存在,则应返回null。例如:

鉴于树:

        4
       / \
      2   7
     / \
    1   3

以及搜索的价值val:2

你应该返回这个子树:

      2     
     / \   
    1   3

在上面的示例中,如果我们要搜索值5,因为没有值为5的节点,我们应该返回null。

注意:空树由null表示,因此你可以将预期输出(序列化树格式)视为[],而不是null。

本次解题使用的开发工具是eclipse,jdk使用的版本是1.8,环境是win7 64位系统,使用Java语言编写和测试。

02 第一种解法

直接使用递归。因为题目给的二叉树是BST,其中序遍历的节点值是从小到大排列且有序的,因此,直接使用中序遍历的顺序,按照左子树–>根–>右子树的顺序遍历即可。

// 借助中序遍历,左根右的顺序
public TreeNode searchBST(TreeNode root, int val) {
    // 先判空
    if (root == null) {
        return null;
    }
    // 如果val大于当前节点值,就往右子树里面找
    if (root.val < val) {
          return searchBST(root.right, val);
    }
    // 如果相等,返回当前以节点为根节点的子树
    if (root.val == val) {
        return root;
    }
    // 如果val小于当前节点值,就往左子树里面找
    if (root.val > val) {
          return searchBST(root.left, val);
    }
    return null;
}

03 第二种解法

也可以世界使用迭代的方式,借助while循环来实现,循环内部的判断逻辑和第一种解法类似,如果当前节点值大于val,就进入左子树找;如果当前节点值小于val,就进入右子树找;如果相等,直接返回以当前节点为根节点的子树。

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;
}

04 第三种解法

我们也可以使用队列来做。将所有节点值依次入队列,在入队列前先判断节点值是否等于val,等于就直接返回当前节点所在子树。

public TreeNode searchBST(TreeNode root, int val) {
    if (root == null || root.val == val) {
        return root;
    }
    Queue<TreeNode> queue = new LinkedList<TreeNode>();
    queue.offer(root);
    while (!queue.isEmpty()) {
        TreeNode temp = queue.poll();
        if (temp.val == val) {
            return temp;
        } 
        if (temp.left != null) {
            queue.offer(temp.left);
        }
        if (temp.right != null) {
            queue.offer(temp.right);
        }
    }
    return null;
}

05 第四种解法

我们也可以将第三种方法再优化,不必所有的节点都进队列,根据节点值的大小,来判断是进左子树还是进右子树,此解法与第二种解法类似,但是使用队列会有点多余,直接使用第二种解法会更好,第二种解法的空间复杂度更低。

public TreeNode searchBST(TreeNode root, int val) {
    if (root == null || root.val == val) {
        return root;
    }
    Queue<TreeNode> queue = new LinkedList<TreeNode>();
    queue.offer(root);
    while (!queue.isEmpty()) {
        TreeNode temp = queue.poll();
        if (temp != null && temp.val == val) {
            return temp;
        } else if (temp != null && temp.val > val) {
            queue.offer(temp.left);
        } else if (temp != null && temp.val < val) {
            queue.offer(temp.right);
        }
    }
    return null;
}

06 小结

算法专题目前已日更超过四个月,算法题文章163+篇,公众号对话框回复【数据结构与算法】、【算法】、【数据结构】中的任一关键词,获取系列文章合集。

以上就是全部内容,如果大家有什么好的解法思路、建议或者其他问题,可以下方留言交流,点赞、留言、转发就是对我最大的回报和支持!

    原文作者:悦乐书
    原文地址: https://www.jianshu.com/p/123427b50510
    本文转自网络文章,转载此文章仅为分享知识,如有侵权,请联系博主进行删除。
点赞