[LeetCode]222. Count Complete Tree Nodes

最新推荐文章于 2024-01-20 11:36:34 发布

原创最新推荐文章于 2024-01-20 11:36:34 发布 · 489 阅读

0 ·

CC 4.0 BY-SA版权

LeetCode 专栏收录该内容

324 篇文章

订阅专栏

本文介绍了一种计算完全二叉树中节点数量的方法。通过判断左右子树是否为满二叉树来递归计算节点总数。文章提供了一个Java实现示例。

https://leetcode.com/problems/count-complete-tree-nodes/

计算完全二叉树中的节点个数

完全二叉树当前结点的左右子树至少有一个是满二叉树

// 完全二叉树树的高度为根节点到最左子树的长度
public class Solution {
    public int countNodes(TreeNode root) {
        // height(root.right) == h - 1表明左子树为满二叉树，再加上root，总共为1 << h，
        // height(root.right) == h - 2表明右子树为满二叉树
        int h = height(root);
        // 位运算全部加括号！！！
        return h < 0 ? 0 : 
            height(root.right) == h - 1 ? (1 << h) + countNodes(root.right) : 
                (1 << (h - 1)) + countNodes(root.left);
    }
    private int height(TreeNode root) {
        if (root == null) {
            return -1;
        }
        return 1 + height(root.left);
    }
}