代码随想录算法训练营第20天|654.最大二叉树、617. 合并二叉树、98. 验证二叉搜索树、700. 二叉搜索树中的搜索

最新推荐文章于 2023-11-12 11:08:54 发布

Albert锐

最新推荐文章于 2023-11-12 11:08:54 发布

阅读量85

点赞数

分类专栏：代码训练营刷题文章标签：算法数据结构

本文链接：https://blog.csdn.net/weixin_43541159/article/details/127998283

版权

代码训练营刷题专栏收录该内容

19 篇文章 0 订阅

订阅专栏

654. 最大二叉树

难度中等600

给定一个不重复的整数数组 nums 。 最大二叉树 可以用下面的算法从 nums 递归地构建:

创建一个根节点，其值为 nums 中的最大值。
递归地在最大值左边的 子数组前缀上 构建左子树。
递归地在最大值右边的 子数组后缀上 构建右子树。

返回 nums 构建的 最大二叉树 。

class Solution {
    public TreeNode constructMaximumBinaryTree(int[] nums) {
        return dfs(nums,0,nums.length-1);
    }
    public TreeNode dfs(int[] nums,int l,int r){
        if(l > r)return null;
        int index = -1;
        int max = Integer.MIN_VALUE;
        for(int i = l;i <= r;i++){
            if(nums[i] > max){
                index = i;
                max = nums[i];
            }
        }
        TreeNode root = new TreeNode(nums[index]);
        root.left = dfs(nums,l,index - 1);
        root.right = dfs(nums,index + 1,r);
        return root;
    }
}

617. 合并二叉树

难度简单1120

给你两棵二叉树： root1 和 root2 。

想象一下，当你将其中一棵覆盖到另一棵之上时，两棵树上的一些节点将会重叠（而另一些不会）。你需要将这两棵树合并成一棵新二叉树。合并的规则是：如果两个节点重叠，那么将这两个节点的值相加作为合并后节点的新值；否则，不为 null 的节点将直接作为新二叉树的节点。

返回合并后的二叉树。

注意: 合并过程必须从两个树的根节点开始。

示例 1：

输入：root1 = [1,3,2,5], root2 = [2,1,3,null,4,null,7]
输出：[3,4,5,5,4,null,7]

class Solution {
    //中序
    public TreeNode mergeTrees(TreeNode root1, TreeNode root2) {
        return dfs(root1,root2);
    }
    public TreeNode dfs(TreeNode p,TreeNode q){
        if(p == null)return q;
        if(q == null)return p;
       TreeNode left  = dfs(p.left,q.left);
        TreeNode root = null;
        if(p != null && q != null){
            root = new TreeNode(p.val + q.val);
        }
        TreeNode right = dfs(p.right,q.right);
        root.left = left;
        root.right = right;
        return root;        
    }
}

98. 验证二叉搜索树

难度中等1818

给你一个二叉树的根节点 root ，判断其是否是一个有效的二叉搜索树。

有效二叉搜索树定义如下：

节点的左子树只包含小于当前节点的数。
节点的右子树只包含大于当前节点的数。
所有左子树和右子树自身必须也是二叉搜索树。

示例 1：

输入：root = [2,1,3]
输出：true

class Solution {
    long pre = Long.MIN_VALUE;
    public boolean isValidBST(TreeNode root) {
        return dfs(root);
    }
    public boolean dfs(TreeNode root){
        if(root == null)return true;
        boolean left = dfs(root.left);
        if(root.val <= pre)return false;
        pre = root.val;
        boolean right = dfs(root.right);
        return left && right;
    }
}

700. 二叉搜索树中的搜索

难度简单340

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

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

示例 1:

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

class Solution {
    public TreeNode searchBST(TreeNode root, int val) {
        return dfs(root,val);
    }
    public TreeNode dfs(TreeNode root,int val){
        if(root == null || root.val == val)return root;
        if(val < root.val)return dfs(root.left,val);
        if(val > root.val)return dfs(root.right,val);
        return null;
    }
}