LeetCode 数据结构入门 day 13 - 树

这是我参与2022首次更文挑战的第13天，活动详情查看：2022首次更文挑战。

二叉搜索树中的搜索

原题地址

题目

给定二叉搜索树（BST）的根节点 root 和一个整数值 val。

你需要在 BST 中找到节点值等于 val 的节点。 返回以该节点为根的子树。 如果节点不存在，则返回 null 。

示例 1:

输入：root = [4,2,7,1,3], val = 2
输出：[2,1,3]

示例 2:

输入：root = [4,2,7,1,3], val = 5
输出：[]

方法

思路：

1. 若 root 为 null，则返回 null；
2. 若 root.val 跟 targetVal 相等, 则返回 root；
3. 递归调用 searchBST 方法处理 root 的 左节点和右节点。

代码：

/**
 * Definition for a binary tree node.
 * function TreeNode(val, left, right) {
 *     this.val = (val===undefined ? 0 : val)
 *     this.left = (left===undefined ? null : left)
 *     this.right = (right===undefined ? null : right)
 * }
 */
/**
 * @param {TreeNode} root
 * @param {number} val
 * @return {TreeNode}
 */
var searchBST = function(root, val) {
    if(!root) return null
    if(root.val === val) {
        return root
    }
    return searchBST(root.left, val) ||  searchBST(root.right, val) || null
};

结果：

• 执行结果： 通过
• 执行用时：72 ms, 在所有 JavaScript 提交中击败了97.03%的用户
• 内存消耗：48.4 MB, 在所有 JavaScript 提交中击败了8.38%的用户
• 通过测试用例：36 / 36

二叉搜索树中的插入操作

原题地址

题目

给定二叉搜索树（BST）的根节点 root 和要插入树中的值 value ，将值插入二叉搜索树。 返回插入后二叉搜索树的根节点。 输入数据 保证 ，新值和原始二叉搜索树中的任意节点值都不同。

注意，可能存在多种有效的插入方式，只要树在插入后仍保持为二叉搜索树即可。 你可以返回 **任意有效的结果 **。

示例 1：

输入：root = [4,2,7,1,3], val = 5
输出：[4,2,7,1,3,5]

示例 2：

输入：root = [40,20,60,10,30,50,70], val = 25
输出：[40,20,60,10,30,50,70,null,null,25]

示例 3：

输入：root = [4,2,7,1,3,null,null,null,null,null,null], val = 5
输出：[4,2,7,1,3,5]

解题方法

思路：

1. 若根节点为 null，则返回一个用 val 构造的 TreeNode；
2. 若 val < root.val，则 左节点 为递归调用 insertIntoBST 返回的值；
3. 若 val > root.val，则 右节点 为递归调用 insertIntoBST 返回的值。

代码：

/**
 * Definition for a binary tree node.
 * function TreeNode(val, left, right) {
 *     this.val = (val===undefined ? 0 : val)
 *     this.left = (left===undefined ? null : left)
 *     this.right = (right===undefined ? null : right)
 * }
 */
/**
 * @param {TreeNode} root
 * @param {number} val
 * @return {TreeNode}
 */
var insertIntoBST = function(root, val) {
    if(!root) return new TreeNode(val)
    if(val < root.val) {
        root.left = insertIntoBST(root.left, val)
    }
    if(val > root.val) {
        root.right = insertIntoBST(root.right, val)
    }
    return root
};

结果：

• 执行结果： 通过
• 执行用时：112 ms, 在所有 JavaScript 提交中击败了67.58%的用户
• 内存消耗：48.8 MB, 在所有 JavaScript 提交中击败了5.05%的用户
• 通过测试用例：35 / 35

— END