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

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

说明：

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

示例:

输入: 
	1
   / \
  2   3
 / \  /
4  5 6

输出: 6

解答

最简单的层次遍历：

/**
 * 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)
            return 0;
        queue<TreeNode*> q;
        q.push(root);
        int result = 0;
        while(!q.empty()){
            int cnt = q.size();
            result += cnt;
            for(int i = 0; i < cnt; i++){
                TreeNode* node = q.front();
                q.pop();
                if(node->left)
                    q.push(node->left);
                if(node->right)
                    q.push(node->right);
            }
        }
        return result;
    }
};

参考官方题解，由于题目规定是完全二叉树，所以可以根据左右子树的深度来简化一半的遍历，即：（1）若左右子树深度相等，则左子树为满子树，其节点个数（加上根节点）数目为$2^{leftDepth}$；（2）若左子树深度小于右子树，则右子树为满子树。对于满子树节点个数直接套公式即可，另一半则可以递归处理。

/**
 * 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)
            return 0;
        int leftDepth = getDepth(root->left);
        int rightDepth = getDepth(root->right);
        // 左子树满
        if(leftDepth == rightDepth){
            return (1 << leftDepth) + countNodes(root->right); 
        }
        // 右子树满
        else{
            return (1 << rightDepth) + countNodes(root->left);
        }
    }

    int getDepth(TreeNode* node){
        if(!node)
            return 0;
        int depth = 0;
        while(node){
            depth += 1;
            node = node->left;
        }
        return depth;
    }
};