代码随想录算法训练营第十六天 | 104.二叉树的最大深度、559.n叉树的最大深度、111.二叉树的最小深度、222.完全二叉树的节点个数
104.二叉树的最大深度
递归法(层序遍历迭代法在第十五天)
思路:本题可以使用前序遍历(中、左、右),也可以使用后序遍历(左、右、中),使用前序求的就是深度,使用后序求的是高度。
- 二叉树节点的深度:指从根节点到该节点的最长简单路径边的条数或者节点数(取决于深度从0开始还是从1开始)
- 二叉树节点的高度:指从该节点到叶子节点的最长简单路径边的条数或者节点数(取决于高度从0开始还是从1开始)
而根节点的高度就是二叉树的最大深度,所以本题通过后序遍历方式求根节点高度来求二叉树最大深度。
class Solution {
public int maxDepth(TreeNode root) { // 递归要素一:确定递归函数的参数和返回值
if (root == null) return 0; // 递归要素二:确定终止条件
// 递归要素三:确定单层递归的逻辑
// 先求左子树的深度,再求右子树的深度,最后取左右深度的最大值后 +1, 即为所求结果
// (+1 是因为算上当前中间节点)
int leftDepth = maxDepth(root.left); // 左
int rightDepth = maxDepth(root.right); // 右
return 1 + (leftDepth > rightDepth ? leftDepth : rightDepth); // 中
}
}
559.n叉树的最大深度
题目链接:559.n叉树的最大深度
思路:与上一题基本类似,递归 + 后续遍历。
class Solution {
public int maxDepth(Node root) {
if (root == null) return 0;
int sum = 0;
if (root.children != null) {
for (Node node : root.children) {
// sum = sum > maxDepth(node) ? sum : maxDepth(node); 会超时
sum = Math.max(sum, maxDepth(node));
}
}
return sum + 1;
}
}
111.二叉树的最小深度(层序遍历迭代法在第十五天)
题目链接:111.二叉树的最小深度
思路:本题依然是前序遍历和后序遍历都可以,前序求的是深度,后序求的是高度。最小深度是从根节点到最近叶子节点的最短路径上的节点数量。
注意:在确定单层递归逻辑时,如果左子树为空,右子树不为空,说明最小深度是 (1 + 右子树的深度)。反之,右子树为空,左子树不为空,最小深度是 (1 + 左子树的深度)。 最后如果左右子树都不为空,返回 (左右子树深度最小值 + 1)。
递归法
class Solution {
public int minDepth(TreeNode root) { // 确定递归函数的参数和返回值
if (root == null) return 0; // 确定终止条件
// 确定单层递归的逻辑
int leftDepth = minDepth(root.left);
int rightDepth = minDepth(root.right);
if (root.left == null && root.right != null) { // 左
return 1 + rightDepth;
}
if (root.left != null && root.right == null) { // 右
return 1 + leftDepth;
}
return 1 + Math.min(leftDepth, rightDepth); // 中
}
}
对比 104 和 111 两题,可见求二叉树的最小深度和求二叉树的最大深度的差别主要在于处理左右孩子不为空的逻辑。
222.完全二叉树的节点个数
普通二叉树求法
思路:仍然采用递归的方式遍历所有节点。
class Solution {
// 通用递归解法
public int countNodes(TreeNode root) {
if(root == null) {
return 0;
}
return countNodes(root.left) + countNodes(root.right) + 1;
}
}
利用完全二叉树性质的求法
思路:在完全二叉树中,除了最底层节点可能没填满外,其余每层节点数都达到最大值,并且最下面一层的节点都集中在该层最左边的若干位置。若最底层为第 h 层,则该层包含 1 ~ 2(h-1)个节点。
完全二叉树只有两种情况,情况一:就是满二叉树,情况二:最后一层叶子节点没有满。
- 对于情况一,可以直接用 2^(树深度 - 1)^ 来计算,注意这里根节点深度为 1。
- 对于情况二,分别递归左孩子和右孩子,递归到某一深度一定会有左孩子或者右孩子为满二叉树,然后依然可以按照情况 1 来计算。
如何判断一个左子树或者右子树是不是满二叉树呢?
在完全二叉树中,如果递归向左遍历的深度等于递归向右遍历的深度,那就说明是满二叉树,如图所示:
在完全二叉树中,如果递归向左遍历的深度不等于递归向右遍历的深度,则说明不是满二叉树,如图所示:
class Solution {
/**
* 针对完全二叉树的解法
*
* 满二叉树的结点数为:2 ^ (depth - 1)
*/
public int countNodes(TreeNode root) {
if (root == null) return 0;
TreeNode left = root.left;
TreeNode right = root.right;
int leftDepth = 0, rightDepth = 0; // 这里初始为 0 是有目的的,为了下面求指数方便
while (left != null) { // 求左子树深度
left = left.left;
leftDepth++;
}
while (right != null) { // 求右子树深度
right = right.right;
rightDepth++;
}
if (leftDepth == rightDepth) {
return (2 << leftDepth) - 1; // 注意(2<<1) 相当于2^2,所以leftDepth初始为 0。
}
return countNodes(root.left) + countNodes(root.right) + 1;
}
}
- 时间复杂度:O(log n × log n)
- 空间复杂度:O(log n)