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

第16天，坚持！

以下题目皆在Carl老师的 代码随想录 中（页面左边有对应题目的目录）有详细讲解，这是博主学习的心得和日志，感兴趣的建议去阅读一下原文。

---

104. 二叉树的最大深度

/**
 * 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 getdepth (TreeNode* node){
        //终止条件
        if ( node == nullptr) 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);
    }
};

---

111. 二叉树的最小深度

/**
 * 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 getDepth(TreeNode* node){
        if (node == nullptr) return 0;
        int leftDepth = getDepth(node -> left);
        int rightDepth = getDepth(node -> right);
        //当左子树为空，右子树不为空
        if (node->left == nullptr && node->right != nullptr){
            return 1+rightDepth;
        }
        //当右子树为空，左子树不为空
        if (node->right == nullptr && node->left != nullptr){
            return 1+leftDepth;
        }

        int res = 1 + min(leftDepth, rightDepth);
        return res;
    }


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

---

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

/**
 * 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) {}
 * };
 */
 #include<cmath>

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;
       while(left){
        left = left->left;
        leftDepth++;
        }
        while(right){
            right = right->right;
            rightDepth++;
        }
        if (rightDepth == leftDepth){
            //左移操作符可以用于快速计算2的幂次方，例如2的n次方可以用左移n位的方式进行计算。
            return(2<<leftDepth) - 1; // 注意(2<<1) 相当于2^2，所以leftDepth初始为0
        }
        return countNodes(root->left) + countNodes(root->right) + 1;
    }
};