最大二叉树
给定一个不含重复元素的整数数组。一个以此数组构建的最大二叉树定义如下:
- 二叉树的根是数组中的最大元素。
- 左子树是通过数组中最大值左边部分构造出的最大二叉树。
- 右子树是通过数组中最大值右边部分构造出的最大二叉树。
通过给定的数组构建最大二叉树,并且输出这个树的根节点。
输入:nums = [3,2,1,6,0,5]
输出:[6,3,5,null,2,0,null,null,1]
构造树一般采用的是前序遍历,因为先构造中间节点,然后递归构造左子树和右子树
递归构造有三个注意事项:
- 确定递归函数的参数和返回值
参数传入的是存放元素的数组,返回该数组构造的二叉树的头结点
- 确定终止条件
当传入数组只有一个元素时,即遍历到了叶子节点。或者数组为空时,返回空
- 确定单层递归的逻辑
单层递归,主要是先找到最大值,然后以这个最大值分割数组,左边是其左子树,右边是右子树
代码
private TreeNode constructMaximumBinaryTree1(int[] nums, int leftIndex, int rightIndex) {
if (rightIndex-leftIndex<1){
return null;
}
if (rightIndex-leftIndex==1){
return new TreeNode(nums[leftIndex]);
}
//先找到最大的元素
int maxIndex=leftIndex;
int maxValue=nums[maxIndex];
for (int i = leftIndex; i < rightIndex; i++) {
if (nums[i]>maxValue){
maxIndex=i;
maxValue=nums[i];
}
}
//最大元素左边是左子树,右边是右子树
TreeNode node = new TreeNode(maxValue);
node.left=constructMaximumBinaryTree1(nums,leftIndex,maxIndex);
node.right=constructMaximumBinaryTree1(nums,maxIndex+1, rightIndex);
return node;
}
合并二叉树
给定两个二叉树,想象当你将它们中的一个覆盖到另一个上时,两个二叉树的一些节点便会重叠。
你需要将他们合并为一个新的二叉树。合并的规则是如果两个节点重叠,那么将他们的值相加作为节点合并后的新值,否则不为 NULL 的节点将直接作为新二叉树的节点。
输入:root1 = [1,3,2,5], root2 = [2,1,3,null,4,null,7]
输出:[3,4,5,5,4,null,7]
确定递归函数的参数和返回值:
传入两个二叉树的根节点,返回值就是合并之后二叉树的根节点
确定终止条件:
如果t1 == NULL 了,两个树合并就应该是 t2 了
反过来如果t2 == NULL,那么两个数合并就是t1
确定单层递归的逻辑:
把两棵树的元素加到一起
代码
public TreeNode mergeTrees2(TreeNode root1, TreeNode root2) {
if (root1==null) return root2;
if (root2==null) return root1;
root1.val+= root2.val;
root1.left=mergeTrees(root1.left,root2.left);
root1.right=mergeTrees(root1.right,root2.right);
return root1;
}
二叉搜索树中的搜索
给定二叉搜索树(BST)的根节点和一个值。 你需要在BST中找到节点值等于给定值的节点。 返回以该节点为根的子树。 如果节点不存在,则返回 NULL
代码
public TreeNode searchBST(TreeNode root, int val) {
if (root==null){
return null;
}
if (root.val==val){
return root;
}
if (root.val<val){
return searchBST(root.right, val);
}
return searchBST(root.left, val);
}