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

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

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

说明：

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

思路

利用完成二叉树的性质，分别求左右子树的高度，如果左右子树的高度相等，证明左子树已经被填满，左子树的节点数为2^n - 1，n为左子树高度，再加上根节点就是2^n，然后递归求解右子树即可。如果左右子树的高度不相等，证明右子树已经没填满，计算右子树的节点数后递归求解左子树即可。

代码

/**
 * Definition for a binary tree node.
 * struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode(int x) : val(x), left(NULL), right(NULL) {}
 * };
 */
class Solution {
public:
    int countNodes(TreeNode* root) {
        if(root == nullptr) {
            return 0;
        }
        int left = countHeight(root -> left);
        int right = countHeight(root -> right);
        if(left == right) {
            return countNodes(root -> right) + (1 << left);
        }
        else {
            return countNodes(root -> left) + (1 << right);
        }
    }
    int countHeight(TreeNode* root) {
        int height = 0;
        while(root != nullptr) {
            height++;
            root = root -> left;
        }
        return height;
    }
};