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

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

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

题解：
方法一：可以通过遍历的方式求解，时间复杂度为O(n)

方法二：
由完全二叉树的性质可以得知，最后一层的最左侧节点一定不为空，因此可以设置指针node一直往左走，来计算完全二叉树的层数。假设层数是从0开始，一共有h层。
那么最后一层的节点编号应该为：2 ^ h ~ 2 ^ (h + 1) - 1。
因此可以采用二分查找的方法，左边界low = 2 ^ h ，右边界high = 2 ^ (h + 1) - 1。
判断节点mid = （low + high + 1） / 2,是否存在，如果存在，证明小于该编号的节点都存在，则low = mid，否则high = mid - 1;

如何判断某一结点是否存在：
例如：判断节点12是否存在，12的二进制表示为：1100.
除了最高位的1，其余各位可以看成从根节点开始，往右走记为1，往左走记为0.判断路径上的节点是否存在即可，如果路径上节点都存在，则该节点存在，否则该节点不存在。

时间复杂度：$O(lo{g}^{2}n)$

/**
 * Definition for a binary tree node.
 * public class TreeNode {
 *     int val;
 *     TreeNode left;
 *     TreeNode right;
 *     TreeNode() {}
 *     TreeNode(int val) { this.val = val; }
 *     TreeNode(int val, TreeNode left, TreeNode right) {
 *         this.val = val;
 *         this.left = left;
 *         this.right = right;
 *     }
 * }
 */
class Solution {
    public int countNodes(TreeNode root) {
        TreeNode node = root;
        int level = 0;
        if(root == null)
            return 0;
        while(node != null){
            level++;
            node = node.left;
        }
        int low = 1 << (level - 1), high = (1 << level) - 1;
        
        while(low < high){
            int mid = (low + high + 1) >> 1;
            if(exits(root,level - 1,mid))
                low = mid;
            else
                high = mid - 1;
        }
        return low;
    }
    public boolean exits(TreeNode root, int level, int k){
        int bits = 1 << (level - 1);
        TreeNode node = root;
        while(node != null && bits > 0){
            if((bits & k) == 0)
                node = node.left;  //往左走为0
            else
                node = node.right; //往右走为1
            bits >>= 1;
        }
        return node != null;
    }

}

题目连接：https://leetcode-cn.com/problems/count-complete-tree-nodes/
题解参考：https://leetcode-cn.com/problems/count-complete-tree-nodes/solution/wan-quan-er-cha-shu-de-jie-dian-ge-shu-by-leetco-2/