LeetCode 222. Count Complete Tree Nodes

最新推荐文章于 2025-05-29 22:51:43 发布

USC_ECE

最新推荐文章于 2025-05-29 22:51:43 发布

阅读量101

点赞数

分类专栏： Tree 文章标签： leetcode

本文链接：https://blog.youkuaiyun.com/qq_31361913/article/details/106954174

版权

Tree 专栏收录该内容

27 篇文章

订阅专栏

本文介绍了一种高效计算二叉树中节点数量的方法。通过递归地检查左右子树高度，判断哪一侧是满二叉树，从而快速计算节点总数。关键在于利用满二叉树的性质和位运算提高效率。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

分别检查左右子树的高度left和right。如果left == right则说明左子树一定是满树，那么左子树加上根节点的node个数为2^left，我们再recursively计算右子树的节点就可以了。如果left > right = left - 1则说明右子树是满树，那么右子树加上根加点的node个数为2^right，我们再recursively计算左子树的节点就可以了。还有一个需要注意的点是移位运算符的优先级低于加法，所以需要加上小括号。代码如下：

/**
 * 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) {
        if(root == null){
            return 0;
        }
        int count = 0;
        int left = height(root.left);
        int right = height(root.right);
        if(left == right){
            count = (1 << (left)) + countNodes(root.right); 
        }else if(left > right){
            count = (1 << (right)) + countNodes(root.left);
        }
        return count;
    }
    
    private int height(TreeNode root){
        if(root == null){
            return 0;
        }
        return Math.max(height(root.left), height(root.right)) + 1;
    }
}