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

二分法

  • 思路:
    • 明确完全二叉树的定义: 一棵深度为k的有n个结点的二叉树,对树中的结点按从上至下、从左到右的顺序进行编号,如果编号为i(1≤i≤n)的结点与满二叉树中编号为i的结点在二叉树中的位置相同,则这棵二叉树称为完全二叉树。
    • 如果定义根节点深度为 0,则每层第一个节点到最后一个节点的取值为:
      • 2\^h, 2\^(h+1) - 1

    • 可以一直向左遍历出二叉树的深度,则可知这棵树的节点个数范围;
    • 然后通过二分法,确认编号节点是否在二叉树上:
      • 在二叉树上,更新左边界:min = mid;
      • 不在二叉树上,更新右边界:max = mid - 1;
    • 判断编号 k 节点是否在这颗树上: 如果第 k 个节点位于第 h 层,则 k 的二进制表示包含 h+1位,其中最高位是 1,其余各位从高到低表示从根节点到第 k 个节点的路径,0 表示移动到左子节点,1 表示移动到右子节点。通过位运算得到第 k 个节点对应的路径,判断该路径对应的节点是否存在,即可判断第 k 个节点是否存在。
cpp 复制代码
/**
 * Definition for a binary tree node.
 * struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode() : val(0), left(nullptr), right(nullptr) {}
 *     TreeNode(int x) : val(x), left(nullptr), right(nullptr) {}
 *     TreeNode(int x, TreeNode *left, TreeNode *right) : val(x), left(left), right(right) {}
 * };
 */
class Solution {
public:
    int countNodes(TreeNode* root) {
        if (nullptr == root) {
            return 0;
        }

        int depth = 0;
        TreeNode* node = root;
        while (node->left != nullptr) {
            depth++;
            node = node->left;
        }

        // 2^depth
        int min = 1 << depth;
        // 2^(depth + 1) - 1
        int max = (1 << (depth + 1)) - 1;

        while (min < max) {
            int mid = (max - min + 1) / 2 + min;
            if (exist(root, depth, mid)) {
                min = mid;
            } else {
                max = mid - 1;
            }
        }

        return min;
    }

private:
    bool exist(TreeNode* root, int depth, int k) {
        int bits = 1 << (depth - 1);
        TreeNode* node = root;
        while (node != nullptr && bits > 0) {
            if (!(bits & k)) {
                node = node->left;
            } else {
                node = node->right;
            }

            bits >>= 1;
        }

        return (node != nullptr);
    }
};
相关推荐
longlong int24 分钟前
【每日算法】Day 17-1:位图(Bitmap)——十亿级数据去重与快速检索的终极方案(C++实现)
开发语言·c++·算法
泛舟起晶浪1 小时前
大衣的旅行--前缀和+二分
数据结构·算法
Run_Teenage1 小时前
C语言 数据结构【双向链表】动态模拟实现
c语言·数据结构·链表
歪~~2 小时前
KMP算法
数据结构·c++·算法
梁下轻语的秋缘2 小时前
每日c/c++题 备战蓝桥杯(小球反弹)[运动分解求解,最大公约数gcd]
c语言·c++·学习·算法·数学建模·蓝桥杯
SiMmming2 小时前
【算法竞赛】状态压缩型背包问题经典应用(蓝桥杯2019A4分糖果)
c++·经验分享·算法·职场和发展·蓝桥杯·动态规划
DexterYttt2 小时前
AT_abc212_d [ABC212D] Querying Multiset
数据结构·c++·算法·优先队列
刻意思考3 小时前
KM算法的时间复杂度,为什么可以降低一个数量级
后端·算法·程序员
橙序员小站3 小时前
探究分布式哈希算法:哈希取模与一致性哈希
算法·架构
飞川0013 小时前
【LeetCode 热题100】45:跳跃游戏 II(详细解析)(Go语言版)
算法·go