【代码随想录】算法训练营 第十六天 第六章 二叉树 Part 3

104. 二叉树的最大深度

题目

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

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

示例：

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

思路

用递归来做，其中心思路是，一个结点的最大深度相当于其左右子树的最大深度加一，这就可以利用递归求得子树深度了。

接下来是递归三部曲：

1. 首先确定返回值和参数，返回值肯定是深度，参数则是二叉树结点；

2. 其次确定递归终止条件，也就是当结点为空时返回0；

3. 最后明确每一次递归要做的事，其实就是按找中心思路返回最大深度。

代码

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);
    }
};

111. 二叉树的最小深度

题目

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

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

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

思路

这道题的中心思路跟上面的一样，都是用递归法，每次取左右子树最小深度加一，不过这里有一个易错点，那就是深度要从叶子结点开始算，所以当遇到一个只有一个子树的结点时，不能记录空的一边，而是递归返回有子树的那边的深度。

代码

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 depth = 1 + min(leftdepth, rightdepth);
        return depth;
    }
    int minDepth(TreeNode* root) {
        return getdepth(root); 
    }
};

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

题目

给你一棵 完全二叉树 的根节点 root ，求出该树的节点个数。

完全二叉树 的定义如下：在完全二叉树中，除了最底层节点可能没填满外，其余每层节点数都达到最大值，并且最下面一层的节点都集中在该层最左边的若干位置。若最底层为第 h 层，则该层包含 1~ 2h 个节点。

普通解法

管他完不完全，反正也是二叉树，既然是二叉树，之前学过的递归和非递归遍历都可以用来求结点数，只需要把原来的结点值入vector变成计数器加一，下面是用非递归前序遍历来做的：

class Solution {
public:
    int countNodes(TreeNode* root) {
        stack<TreeNode*> st;
        int num = 0;
        if (root != NULL) st.push(root);
        while (!st.empty()) {
            TreeNode* cur = st.top();
            st.pop();
            num++;
            if (cur->right) st.push(cur->right);
            if (cur->left) st.push(cur->left);
        }
        return num;
    }
};

完全二叉树解法

完全二叉树可以不断拆分为一个满二叉树和一个完全二叉树，每一个满二叉树可以一边往下走一边求深度，如果左右子树的深度相同，就可以直接计算出这个二叉树的结点个数，因为满二叉树的结点数等于2的深度次方减一，如果不是满二叉树，就递归求解这个完全二叉树的根节点的左右子树的结点数再加一。

class Solution {
public:
    int countNodes(TreeNode* root) {
        if (root == NULL) 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 (leftDepth == rightDepth) {
            return (2 << leftDepth) - 1;
        }
        return countNodes(root->left) + countNodes(root->right) + 1;
    }
};