代码随想录算法训练营第十五天（二）| 404.左叶子之和 222.完全二叉树的节点个数

404.左叶子之和

题目：

给定二叉树的根节点 root ，返回所有左叶子之和。

示例 1：

输入: root = [3,9,20,null,null,15,7] 
输出: 24 
解释: 在这个二叉树中，有两个左叶子，分别是 9 和 15，所以返回 24

示例 2:

输入: root = [1]
输出: 0

提示:

• 节点数在 [1, 1000] 范围内
• -1000 <= Node.val <= 1000

思路：

要计算二叉树中所有左叶子节点的和，可以使用递归遍历树的方法。

在遍历过程中，我们检查每个节点的左子节点是否是叶子节点（即没有左右子节点），如果是，则将其值累加到结果中。

1. 使用递归函数 sumOfLeftLeavesHelper 来遍历二叉树的每个节点。
2. 对于每个节点，检查其左子节点是否是叶子节点。如果是，则将其值累加到结果中。
3. 继续递归遍历左子树和右子树。
4. 最后返回累加的左叶子节点的和。

上代码：

/**
 * 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 sumOfLeftLeavesHelper(TreeNode* node) {
        if (!node) return 0;  // 如果节点为空，返回0

        int sum = 0;

        // 检查左子节点是否是叶子节点
        if (node->left && !node->left->left && !node->left->right) {
            sum += node->left->val;
        }

        // 递归遍历左子树和右子树，并累加结果
        sum += sumOfLeftLeavesHelper(node->left);
        sum += sumOfLeftLeavesHelper(node->right);

        return sum;
    }

    // 主函数：返回所有左叶子节点之和
    int sumOfLeftLeaves(TreeNode* root) {
        return sumOfLeftLeavesHelper(root);
    }
    
};

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

题目：

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

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

示例 1：

输入：root = [1,2,3,4,5,6]
输出：6

示例 2：

输入：root = []
输出：0

示例 3：

输入：root = [1]
输出：1

提示：

• 树中节点的数目范围是[0, 5 * 104]
• 0 <= Node.val <= 5 * 104
• 题目数据保证输入的树是 完全二叉树

思路：

要计算一棵完全二叉树的节点数，可以利用完全二叉树的特性来优化计算过程。完全二叉树的特性是：如果树的左子树高度等于右子树高度，那么左子树是满的，可以直接通过公式计算节点数；否则，需要递归计算。

1. 完全二叉树的性质：

   • 如果一棵树的高度为 h，那么最左边的一条路径（即从根节点到最左叶子节点）长度为 h，我们可以通过计算最左路径和最右路径的高度来判断子树的情况。
   • 如果左子树的高度等于右子树的高度，那么左子树一定是满的，节点数可以直接通过 2^h - 1 公式计算，加上右子树的节点数即可。
   • 如果左子树的高度不等于右子树的高度，那么右子树一定是满的，递归计算左子树的节点数。

2. 递归计算：

   • 对于当前节点，计算左子树和右子树的高度，然后根据完全二叉树的性质选择递归的方向。

上代码：

/**
 * 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 getDepthLeft(TreeNode* node) {
        int depth = 0;
        while (node) {
            node = node->left;
            depth++;
        }
        return depth;
    }

    // 计算从节点到最右端叶子节点的高度
    int getDepthRight(TreeNode* node) {
        int depth = 0;
        while (node) {
            node = node->right;
            depth++;
        }
        return depth;
    }

    // 主函数：返回完全二叉树的节点数
    int countNodes(TreeNode* root) {
        if (!root) return 0;  // 如果根节点为空，节点数为0

        int leftDepth = getDepthLeft(root);
        int rightDepth = getDepthRight(root);

        if (leftDepth == rightDepth) {
            // 如果左右高度相同，左子树是满的，节点数为2^leftDepth - 1，加上根节点，再加上右子树节点数
            return pow(2, leftDepth) - 1;
        } else {
            // 如果左右高度不同，递归计算左子树和右子树的节点数
            return 1 + countNodes(root->left) + countNodes(root->right);
        }
    }
    
};