NC 完全二叉树结点数

系列文章目录

---

---

前言

前些天发现了一个巨牛的人工智能学习网站，通俗易懂，风趣幽默，忍不住分享一下给大家。点击跳转到网站，这篇文章男女通用，看懂了就去分享给你的码吧。

---

描述
给定一棵完全二叉树的头节点head，返回这棵树的节点个数。

完全二叉树指：设二叉树的深度为h，则 [1,h-1] 层的节点数都满足个

class Solution {
public:
    int nodeNum(struct TreeNode* head) {
        if (!head) return 0; // 空树算0个
        // 计算左右孩子, 不用判空, 如果left或right为空, 函数返回0
        int lh = getHeight(head->left), rh = getHeight(head->right);

        int res = 1; // 根节点算1个
        if (lh == rh) {
            // (1<<lh) - 1 是左子树满二叉树的节点数, 右子树一定也是完全二叉树, 故递归计算
            res += nodeNum(head->right) + (1<<lh) - 1;
        } else {
            // 同上, 右子树一定是满二叉树
            res += nodeNum(head->left) + (1<<rh) - 1;
        }
        return res;
    }

    int getHeight(struct TreeNode* head) {
        int h = 0; // 初始高度是0
        while (head) {
            h++; // 当前节点存在, 高度加1
            head = head->left; // 向左走
        }
        return h;
    }
};