0111.二叉树的最小深度.md

题目地址

https://leetcode-cn.com/problems/minimum-depth-of-binary-tree/

思路

这道题目建议和0104.二叉树的最大深度一起来做。

来来来，一起递归三部曲：

• 确定递归函数的参数和返回值
• 确定终止条件
• 确定单层递归的逻辑

1. 确定递归函数的参数和返回值

参数为要传入的二叉树根节点，返回的是int类型的深度。

代码如下：

int getDepth(TreeNode* node)

2. 确定终止条件

终止条件也是遇到空节点返回0，表示当前节点的高度为0。

代码如下：

if (node == NULL) return 0;

2. 确定单层递归的逻辑

首先取左右子树的深度，如果左子树为空，右子树不为空，说明最小深度是 1 + 右子树的深度。

反之，右子树为空，左子树不为空，最小深度是 1 + 左子树的深度。 最后如果左右子树都不为空，返回左右子树深度最小值 + 1 就可以了。

可以看出：求二叉树的最小深度和求二叉树的最大深度的差别主要在于处理左右孩子不为空的逻辑。

代码如下：

int leftDepth = getDepth(node->left);
int rightDepth = getDepth(node->right);
if (node->left == NULL && node->right != NULL) { 
    return 1 + rightDepth;
}   
if (node->left != NULL && node->right == NULL) { 
    return 1 + leftDepth;
}
return 1 + min(leftDepth, rightDepth);

整体递归代码如下：

C++代码

递归

class Solution {
public:
    int getDepth(TreeNode* node) {
        if (node == NULL) return 0;
        // 当一个左子树为空，右不为空，这时并不是最低点
        if (node->left == NULL && node->right != NULL) {
            return 1 + getDepth(node->right);
        }
        // 当一个右子树为空，左不为空，这时并不是最低点
        if (node->left != NULL && node->right == NULL) {
            return 1 + getDepth(node->left);
        }
        return 1 + min(getDepth(node->left), getDepth(node->right));
    }

    int minDepth(TreeNode* root) {
        return getDepth(root);
    }
};

迭代法

可以使用二叉树的广度优先遍历（层序遍历），来做这道题目，其实这就是一道模板题了，二叉树的层序遍历模板在这道题解中0102.二叉树的层序遍历

层序遍历如图所示：

代码如下：（详细注释）

class Solution {
public:

    int minDepth(TreeNode* root) {
        if (root == NULL) return 0;
        int depth = 0;
        queue<TreeNode*> que;
        que.push(root);
        while(!que.empty()) {
            int size = que.size(); // 必须要这么写，要固定size大小
            depth++;
            int flag = 0;
            for (int i = 0; i < size; i++) {
                TreeNode* node = que.front();
                que.pop();
                if (node->left) que.push(node->left);
                if (node->right) que.push(node->right);
                if (!node->left && !node->right) { // 当左右孩子都为空的时候，说明是最低点的一层了，退出
                    flag = 1;
                    break;
                }
            }
            if (flag == 1) break;
        }
        return depth;
    }
};

更多算法干货文章持续更新，可以微信搜索「代码随想录」第一时间围观，关注后，回复「Java」「C++」 「python」「简历模板」「数据结构与算法」等等，就可以获得我多年整理的学习资料。