代码随想录算法训练营第十六天【二叉树part03】 | 104.二叉树的最大深度、111.二叉树的最小深度、222.完全二叉树的节点个数

104.二叉树的最大深度

题目链接：

力扣（LeetCode）官网 - 全球极客挚爱的技术成长平台

求解思路：

使用后序遍历。

递归三部曲：

1. 确定递归函数的参数和返回值：参数为根节点，返回值为最大高度，是一个int类型
2. 确定终止条件：如果节点为空的话，就返回0
3. 确定单层递归的逻辑：先求左子树深度，再求右子树深度，最后取左右深度中较大的数加一（加一是因为算上当前的中间节点）。

代码：

/**
 * Definition for a binary tree node.
 * struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode() : val(0), left(nullptr), right(nullptr) {}
 *     TreeNode(int x) : val(x), left(nullptr), right(nullptr) {}
 *     TreeNode(int x, TreeNode *left, TreeNode *right) : val(x), left(left), right(right) {}
 * };
 */
class Solution {
public:
    int maxDepth(TreeNode* root) {
        // 终止条件
        if (root == NULL) return 0;
        // 后序遍历，左右中
        int left = maxDepth(root->left);
        int right = maxDepth(root->right);
        int sum = 1 + max(left, right);
        return sum;
    }
};

111.二叉树的最小深度

题目链接：

力扣（LeetCode）官网 - 全球极客挚爱的技术成长平台

求解思路：

本题有一个误区，如图所示，因此需要特殊处理一下。

递归三部曲：

1. 确定递归函数的参数和返回值：参数为二叉树的根节点，返回值为一个int类型
2. 确定终止条件：遇到空节点返回0
3. 确定单层递归的逻辑：如果左子树为空，右子树不为空，说明最小深度是 1 + 右子树的深度；如果右子树为空，左子树不为空，说明最小深度是 1 + 左子树的深度；除上述情况外，最小深度为 1 + min（左子树最小深度，右子树最小深度）

代码：

/**
 * Definition for a binary tree node.
 * struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode() : val(0), left(nullptr), right(nullptr) {}
 *     TreeNode(int x) : val(x), left(nullptr), right(nullptr) {}
 *     TreeNode(int x, TreeNode *left, TreeNode *right) : val(x), left(left), right(right) {}
 * };
 */
class Solution {
public:
    int minDepth(TreeNode* root) {
        // 终止条件
        if (root == NULL) return 0;
        // 单层逻辑
        int left = minDepth(root->left);
        int right = minDepth(root->right);
        // 处理特殊情况
        if (root->left == NULL && root->right != NULL)
            return 1 + right;
        else if (root->right == NULL && root->left != NULL)
            return 1 + left;
        else
            return 1 + min(left, right);
    }
};

222.完全二叉树的节点个数

题目链接：

力扣（LeetCode）官网 - 全球极客挚爱的技术成长平台

求解思路：

利用完全二叉树的特性，如图所示，如果是满二叉树，可以直接返回节点数量为2^k-1，k为深度。

递归三部曲：

1. 确定递归函数的参数和返回值：参数为二叉树的根节点，返回值为一个int类型
2. 确定终止条件：遇到空节点返回0；如果是满二叉树返回2^k-1
3. 确定单层递归的逻辑：后序遍历，即左右中

代码：

/**
 * Definition for a binary tree node.
 * struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode() : val(0), left(nullptr), right(nullptr) {}
 *     TreeNode(int x) : val(x), left(nullptr), right(nullptr) {}
 *     TreeNode(int x, TreeNode *left, TreeNode *right) : val(x), left(left), right(right) {}
 * };
 */
class Solution {
public:
  int countNodes(TreeNode* root) {
    // 终止条件1，遇到空
    if (root == NULL) return 0;
    // 终止条件2，向左遍历深度等于向右遍历深度（说明为满二叉树）
    TreeNode* left = root->left;
    TreeNode* right = root->right;
    int leftDepth = 0, rightDepth = 0;
    while (left){
        left = left->left;
        leftDepth++;
    }
    while (right){
        right = right->right;
        rightDepth++;
    }
    if (leftDepth == rightDepth){
        return (2<<leftDepth)-1; // 2的leftDepth次方-1
    }
    // 单层递归逻辑
    int leftNum = countNodes(root->left);
    int rightNum = countNodes(root->right);
    return (leftNum + rightNum)+1;
    }
};