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

一、104.二叉树的最大深度 

题目链接/文章讲解/视频讲解：104.二叉树的最大深度

 思考：通过后序求的根节点高度来求的二叉树最大深度

代码实现： 

class Solution {
public:
    int getdepth(TreeNode* node) {
        if (node == NULL) return 0;
        int leftdepth = getdepth(node->left);       // 左
        int rightdepth = getdepth(node->right);     // 右
        int depth = 1 + max(leftdepth, rightdepth); // 中
        return depth;
    }
    int maxDepth(TreeNode* root) {
        return getdepth(root);
    }
};

精简版：

class solution {
public:
    int maxDepth(TreeNode* root) {
        if (root == null) return 0;
        return 1 + max(maxDepth(root->left), maxDepth(root->right));
    }
};

二、111.二叉树的最小深度

题目链接/文章讲解/视频讲解：111.二叉树的最小深度

思考：

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

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

代码实现： 

class Solution {
public:
    int getDepth(TreeNode* node) {
        if (node == NULL) return 0;
        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;
        }
        int result = 1 + min(leftDepth, rightDepth);
        return result;
    }

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

精简版：

class Solution {
public:
    int minDepth(TreeNode* root) {
        if (root == NULL) return 0;
        if (root->left == NULL && root->right != NULL) {
            return 1 + minDepth(root->right);
        }
        if (root->left != NULL && root->right == NULL) {
            return 1 + minDepth(root->left);
        }
        return 1 + min(minDepth(root->left), minDepth(root->right));
    }
};

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

题目链接/文章讲解/视频讲解：222.完全二叉树的节点个数

思考：

1.题目要求的是完全二叉树，因此要充分利用二叉树的特性。

2.完全二叉树只有两种情况，情况一：就是满二叉树，情况二：最后一层叶子节点没有满。

        情况一，可以直接用 2^树深度 - 1 来计算，其中这里根节点深度为1。

        情况二，分别递归左孩子，和右孩子，递归到某一深度一定会有左孩子或者右孩子为满二叉树，然后依然可以按照情况1来计算。

3.1普通二叉树解法

class Solution {
private:
    int getNodesNum(TreeNode* cur) {
        if (cur == NULL) return 0;
        int leftNum = getNodesNum(cur->left);      // 左
        int rightNum = getNodesNum(cur->right);    // 右
        int treeNum = leftNum + rightNum + 1;      // 中
        return treeNum;
    }
public:
    int countNodes(TreeNode* root) {
        return getNodesNum(root);
    }
};

• 时间复杂度：O(n)
• 空间复杂度：O(log n)

3.2完全二叉树解法

class Solution {
public:
    int countNodes(TreeNode* root) {
        if (root == nullptr) return 0;
        TreeNode* left = root->left;
        TreeNode* right = root->right;
        int leftDepth = 0, rightDepth = 0; // 这里初始为0是有目的的，为了下面求指数方便

        while (left) // 求左子树深度
        {  
            left = left->left;
            leftDepth++;
        }
        while (right) // 求右子树深度
        { 
            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)