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

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

题目

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

说明：

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

示例:

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

输出: 6

思路

最直接的思路是统计左子树节点的数量，再统计右子树节点的数量，然后加起来，时间复杂度是O（N）的。但是这不是最优的思路
最优的思路是：充分利用完全二叉树的性质。

首先统计根节点的高度，设为h。接着统计root.right节点的高度h1，如果h1+1 = h，说明此时，根节点的左子树，是一个满二叉树，那么左子树的节点数量就是2^(h-1)-1，加上根节点的数量，就是2 ^(h-1)。此时只需要统计右子树的节点数量，加在一起就是最终结果。如下图所示

如果h1+1 <h ，令t = h1+1，说明此时右子树是满的，那么右子树和根节点的节点数量，加在一起就是2 ^(t-1)，此时统计左子树的节点数量，加一起就可以了。如下图所示

时间复杂度：第一次需要统计树的高度，时间复杂度是O(d)，第二次需要统计左子树或者右子树的高度O(d-1)，…，加在一起，O(d^2)，d是树的高度（logN)。

代码

这里直接上最优解的代码

//求高度
    public int depth(TreeNode root,int h){
        if(root!=null){
            return depth(root.left,h+1);
        }else{
            return h;
        }
    }
    public int countNodes(TreeNode root) {
        int h = depth(root,0);
        if(h==1 || h==0){
            return h;
        }
        if(root.right!=null){
            int t = depth(root.right,1);
            if(h==t){
                int t1 = 1<<(t-1);
                int t2 = countNodes(root.right);
                return t1+t2;
            }else{
                int t1 = 1<<(t-1);
                int t2 = countNodes(root.left);
                return t1+t2;
            }
        }else{
            return 2;
        }

       
    }