LeetCode 第 222 题：完全二叉树的节点个数(C++)

222. 完全二叉树的节点个数 - 力扣（LeetCode）

层次遍历

class Solution {
public:
    int countNodes(TreeNode* root) {
        if(!root)   return 0;
        queue<TreeNode*> q;
        q.push(root);
        int res = 0, cnt = 1;
        while(!q.empty()){
            int size = q.size();
            if(size == cnt) res += cnt;
            else{
                res += size;    break;
            }
            cnt *= 2;
            for(int i = 0; i < size; ++i){
                auto cur = q.front();
                q.pop();
                if(cur->left)   q.push(cur->left);
                if(cur->right)  q.push(cur->right);
            }
        }
    	return res;
    }
};

递归

class Solution {
public:
    int countNodes(TreeNode* root) {
        if(!root)   return 0;
        return countNodes(root->left) + countNodes(root->right) + 1;
    }
};

二分+位运算

但是其实上面两种方法都没有利用到完全二叉树的性质，我看了官方题解之后惊了，这也能用二分（虽然效率上并没有提升太多）：

根据完全二叉树的性质，从根节点一直往左走，直到遇到叶子节点，经过的路径长度就是最大层数 h（层数从0开始计算），那么第 h 层的节点数必然在区间[1, 2^h]之内。

所以就可以在区间[1, 2^h]二分来寻找节点数，二分的思路就更巧了：

假设需要判断的节点数为 k，如果

• 第 k个节点存在：节点个数 >= k
• 第 k个节点不存在：节点个数 < k

所以重点就在于如何去判断第 k个节点是否存在，后者如何去定位第 k个节点，思路是将二叉树与二进制相互结合，源自完全二叉树的节点个数 - 完全二叉树的节点个数 - 力扣（LeetCode）：

如果第 k 个节点位于第 h 层，则 k 的二进制表示包含 h+1 位，其中最高位是 1，其余各位从高到低表示从根节点到第 k 个节点的路径，0 表示移动到左子节点，1 表示移动到右子节点。通过位运算得到第 k 个节点对应的路径，判断该路径对应的节点是否存在，即可判断第 k 个节点是否存在。

class Solution {
public:
    int countNodes(TreeNode* root) {
        if(!root)   return 0;
        int h = 0; //层数
        auto p = root;
        while(p->left){//一直往左走得到层数h
            ++h;
            p = p->left;
        }
        if(h == 0)  return 1;
        int low = 1 << h, high = (1 << (h+1)) - 1;
        while(low <= high){
            int mid = (low + high)/2;
            if(exist(root, h, mid)){
                if(mid == high || !exist(root, h, mid+1))    return mid;
                else low = mid+1;
            }else
                high = mid - 1;
        }
        return 0;
    }
    bool exist(TreeNode *root, int h, int k){
        auto p = root;
        int bits = 1 << (h-1);//从高位到低位，先取的是数字k的二进制的第二高位（第一高位为1）
        while(p && bits > 0){
            if(bits & k)   p = p->right;
            else p = p->left;
            bits >>= 1;
        }
        return p != NULL;
    }
};

效率上其实并没有提升很多，但是思路真的很巧妙。