代码随想录算法训练营第二十天|654.最大二叉树、617.合并二叉树、700.二叉搜索树中的搜索、98.验证二叉搜索树
654.最大二叉树
问题简述:每次将最大元素作为当前结点,并将最大元素的左右侧元素分别作为左右子树。
思考:还是不需要创建新的数组,只需要创建左右指针。和前序中序构造树有点相似。
算法思路:定义双指针分别指向数组头尾,找出双针内范围最大值。最为当前节点,然后递归遍历最大值左面的元素作为左子树,最大值右面的元素作为右子树,最后返回当前节点。
class Solution {
public TreeNode constructMaximumBinaryTree(int[] nums) {
return constructMaximum(nums, 0, nums.length - 1);
}
public TreeNode constructMaximum(int[] nums, int left, int right){
//取区间范围为左闭右闭
if (left > right) return null;
//范围内只有一个结点,直接返回该结点
if (left == right) return new TreeNode(nums[left]);
int max = Integer.MIN_VALUE;
//记录最大值下标
int index = 0;
for (int i = left; i <= right; i++) {
if (nums[i] > max){
index = i;
max = nums[i];
}
}
TreeNode root = new TreeNode(max);
root.left = constructMaximum(nums, left, index - 1);
root.right = constructMaximum(nums, index + 1, right);
return root;
}
}
617.合并二叉树
问题简述:将两个二叉树进行合并,重叠的节点进行加合,不重叠的节点直接建立。
思考:自己写的创建了新的结点来存储二叉树,其实不用,直接将r2放入r1中就好。
算法思路:同时遍历两个结点,如果都为空结点,返回null。如果不都为空说明存在结点,则创建一个新结点,并为其赋值,再递归链接当前节点的左右子树。最后返回新建的结点。
class Solution {
public TreeNode mergeTrees(TreeNode root1, TreeNode root2) {
if (root1 == null && root2 == null){
return null;
}
TreeNode root = new TreeNode();
if (root1 == null && root2 != null) root.val = root2.val;
if (root1 != null && root2 == null) root.val = root1.val;
if (root1 != null && root2 != null) root.val = root1.val + root2.val;
//如果左右子树不为空,则递归链接左右子树
root.left = mergeTrees((root1 != null) ? root1.left : null, (root2 != null) ? root2.left : null);
root.right = mergeTrees((root1 != null) ? root1.right : null, (root2 != null) ? root2.right : null);
return root;
}
}
改进:访问当前两个结点,如果一个为空,直接返回另一个结点;都不为空直接修改root1的值为加和。连接左右结点,最后返回root1。
class Solution {
public TreeNode mergeTrees(TreeNode root1, TreeNode root2) {
//如果一个为空,直接返回另一个
if (root1 == null) return root2;
if (root2 == null) return root1;
//都不为空直接修改root1的值
root1.val = root1.val + root2.val;
//链接左右结点
root1.left = mergeTrees(root1.left, root2.left);
root1.right = mergeTrees(root1.right, root2.right);
//因为修改的为root1,直接返回root1
return root1;
}
}
700.二叉搜索树中的搜索
问题简述:在二叉搜索树中查找对应某结点的值。
思考:这个比较简单。
算法思路:判断当前节点是否是目标值,如果不是,递归查找左孩子和右孩子,如果最后没找到返回null。
class Solution {
public TreeNode searchBST(TreeNode root, int val) {
if (root == null) return null;
if (val == root.val) return root;
if (val < root.val) return searchBST(root.left, val);
if (val > root.val) return searchBST(root.right, val);
return null;
}
}
98.验证二叉搜索树
问题简述:验证一个树是否为二叉搜索树。
思考:这个开始感觉不难,但是废了好大劲,主要是想的不对。然后知道了应该判断中序遍历序列后,还是不太会直接递归出来,后来看了视频终于懂了,然后这个题好像还可以双指针,回头再看。
算法思路:定义一个long作为上一个结点的值,然后中序遍历二叉树,遍历到当前节点时,判断是否大于等于上一个结点,如果是则返回fasle;不是则更新上一个结点值。最后返回左右子树的boolean交集。
class Solution {
//记录上一个结点的值
long lastnum = Long.MIN_VALUE;
public boolean isValidBST(TreeNode root) {
if (root == null) return true;
//中序遍历
boolean left = isValidBST(root.left);
//判断是否大于上个结点
if (lastnum >= root.val){
return false;
}else {
//更新结点值
lastnum = root.val;
}
boolean right =isValidBST(root.right);
//如果左右子树都满足,则返回true
return left && right;
}
}
感想
继续抓紧吧。