代码随想录算法训练营第十四天（二）|104.二叉树的最大深度 111.二叉树的最小深度

104.二叉树的最大深度

题目：

给定一个二叉树 root ，返回其最大深度。

二叉树的 最大深度 是指从根节点到最远叶子节点的最长路径上的节点数。

示例 1：

输入：root = [3,9,20,null,null,15,7]
输出：3

示例 2：

输入：root = [1,null,2]
输出：2

提示：

• 树中节点的数量在 [0, 104] 区间内。
• -100 <= Node.val <= 100

思路：

首先检查根节点是否为空：如果根节点为空，说明这是一个空树，因此最大深度为0。

如果根节点不为空，我们需要递归地计算左子树和右子树的最大深度：

最后，我们需要比较左右子树的最大深度，并返回较大的那个值加1（加上当前节点）：

上代码：

class Solution {
public:
    // 主函数，计算二叉树的最大深度
    int maxDepth(TreeNode* root) {
        // 如果根节点为空，则深度为0
        if (root == NULL) {
            return 0;
        }
        // 递归计算左子树和右子树的最大深度
        int leftDepth = maxDepth(root->left);
        int rightDepth = maxDepth(root->right);
        // 返回左右子树中较大的深度加1（加上当前节点）
        return max(leftDepth, rightDepth) + 1;
    }
};

111.二叉树的最小深度

题目：

给定一个二叉树，找出其最小深度。

最小深度是从根节点到最近叶子节点的最短路径上的节点数量。

说明：叶子节点是指没有子节点的节点。

示例 1：

输入：root = [3,9,20,null,null,15,7]
输出：2

示例 2：

输入：root = [2,null,3,null,4,null,5,null,6]
输出：5

思路：

这题和最大深度的区别主要在于求解的目标不同。

最大深度是求从根节点到最远叶子节点的最长路径上的节点数量，而最小深度是求从根节点到最近叶子节点的最短路径上的节点数量。

在求解过程中，两者都需要遍历整个树，但最小深度需要在遍历过程中记录当前找到的最短路径长度，并在找到更短的路径时更新这个值。而最大深度则需要记录当前找到的最长路径长度，并在找到更长的路径时更新这个值。

• 分别判断左右孩子是否存在
• 左孩子不存在就返回右孩子并将节点数加1
• 右孩子不存在就返回左孩子并将节点数加1
• 若左右孩子都存在就返回分别返回左右孩子的最小值

上代码：

int minDepth(TreeNode* root) {
    if (root == NULL) return 0; // 如果根节点为空，返回深度为0
    queue<TreeNode*> q; // 创建一个队列用于层序遍历
    q.push(root); // 将根节点加入队列
    int depth = 1; // 初始化深度为1
    while (!q.empty()) { // 当队列不为空时，继续遍历
        int size = q.size(); // 获取当前层的节点数量
        for (int i = 0; i < size; i++) { // 遍历当前层的所有节点
            TreeNode* node = q.front(); // 取出队首节点
            q.pop(); // 弹出队首节点
            if (node->left == NULL && node->right == NULL) return depth; // 如果该节点是叶子节点，返回当前深度
            if (node->left != NULL) q.push(node->left); // 将左子节点加入队列
            if (node->right != NULL) q.push(node->right); // 将右子节点加入队列
        }
        depth++; // 遍历完一层后，深度加1
    }
    return depth; // 返回最小深度
}