代码随想录算法训练营第二十二天 |最大二叉树、合并二叉树 、二叉搜索树中的搜索、验证二叉搜索树
今天的题目全是使用递归完成的,注意递归的三部曲。
最大二叉树link
题目: 给定一个不重复的整数数组 nums 。 最大二叉树 可以用下面的算法从 nums 递归地构建:
创建一个根节点,其值为 nums 中的最大值。
递归地在最大值 左边 的 子数组前缀上 构建左子树。
递归地在最大值 右边 的 子数组后缀上 构建右子树。
返回 nums 构建的 最大二叉树 。
视频推荐:link
class Solution {
public TreeNode constructMaximumBinaryTree(int[] nums) {
return function(nums);
}
//递归
// 先序遍历
// 左闭右开
TreeNode function(int[] nums) {
if (nums.length == 1)
return new TreeNode(nums[0]);
//中
int Maxvalue = -1;
int index = -1;
for (int i = 0; i < nums.length; i++) {
if (nums[i] > Maxvalue) {
Maxvalue = nums[i];
index = i;
}
}
TreeNode root = new TreeNode(Maxvalue);
//左
if (index > 0) {//表示有左子树
int[] newLeft = Arrays.copyOfRange(nums, 0, index);
root.left = function(newLeft);
}
//右
if (index < nums.length - 1) {//表示有右子树
int[] newtRight = Arrays.copyOfRange(nums, index + 1, nums.length);
root.right = function(newtRight);
}
return root;
}
}
合并二叉树link
题目:给你两棵二叉树: root1 和 root2 。
想象一下,当你将其中一棵覆盖到另一棵之上时,两棵树上的一些节点将会重叠(而另一些不会)。你需要将这两棵树合并成一棵新二叉树。合并的规则是:如果两个节点重叠,那么将这两个节点的值相加作为合并后节点的新值;否则,不为 null 的节点将直接作为新二叉树的节点。
返回合并后的二叉树。
注意: 合并过程必须从两个树的根节点开始。
视频推荐:link
class Solution {
public TreeNode mergeTrees(TreeNode root1, TreeNode root2) {
return function(root1, root2);
}
//递归
//先序遍历
//在root1的基础上操作,不适用新的二叉树了
TreeNode function(TreeNode r1, TreeNode r2) {
//结束条件
if (r1 == null)
return r2;
if (r2 == null)
return r1;
//中
r1.val += r2.val;
//左
r1.left = function(r1.left, r2.left);
//右
r1.right = function(r1.right, r2.right);
return r1;
}
}
二叉搜索树中的搜索link
题目:给定二叉搜索树(BST)的根节点 root 和一个整数值 val。
你需要在 BST 中找到节点值等于 val 的节点。 返回以该节点为根的子树。 如果节点不存在,则返回 null 。
视频推荐:link
class Solution {
public TreeNode searchBST(TreeNode root, int val) {
//结束条件
if (root == null || root.val == val)
return root;
TreeNode res = null;
if (root.val > val)
res = searchBST(root.left, val);
if (root.val < val)
res = searchBST(root.right, val);
return res;
}
}
验证二叉搜索树link
题目:给你一个二叉树的根节点 root ,判断其是否是一个有效的二叉搜索树。
有效 二叉搜索树定义如下:
节点的左子树只包含 小于 当前节点的数。
节点的右子树只包含 大于 当前节点的数。
所有左子树和右子树自身必须也是二叉搜索树。
class Solution {
List<Integer> ans = new ArrayList<>();
public boolean isValidBST(TreeNode root) {
traverse(root);
for (int i = 1; i < ans.size(); i++)
if (ans.get(i - 1) >= ans.get(i))
return false;
return true;
}
// 中序遍历,放入数组中,要是符合条件,则数组递增
void traverse(TreeNode root) {
// 结束条件
if (root == null)
return;
// 单层递归逻辑
traverse(root.left);
ans.add(root.val);
traverse(root.right);
}
}
希望对您有帮助,谢谢。