力扣labuladong一刷day40天计算完全二叉树节点数

最新推荐文章于 2025-11-02 11:27:35 发布

原创最新推荐文章于 2025-11-02 11:27:35 发布 · 495 阅读

9 ·

CC 4.0 BY-SA版权

文章标签：

#leetcode #算法 #职场和发展

力扣算法题专栏收录该内容

237 篇文章

订阅专栏

本文介绍了一种在LeetCode题目中计算完全二叉树节点个数的高效算法，利用完全二叉树特性，通过比较左右子树深度，避免了全遍历，提高了效率。

力扣labuladong一刷day40天计算完全二叉树节点数

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

题目链接：https://leetcode.cn/problems/count-complete-tree-nodes/
思路：计算完全二叉树直接全遍历的话很浪费时间，但是可以利用完全二叉树的特性来解题，每到一个节点我们就计算它的left = node.left深度，和right=node.right的深度，如果这两个深度想等，总节点的个数就是2的h次方-1，如果不相等那就把当前节点计数，然后递归左右子树执行上面的，每次左右子树都会有一个可以大化小，小化无。

class Solution {
    public int countNodes(TreeNode root) {
        if (root == null) return 0;
        TreeNode l = root, r = root;
        int a = 0, b = 0;
        while (l != null) {
            l = l.left;
            a++;
        }
        while (r != null) {
            r = r.right;
            b++;
        }
        if (a == b) {
            return (int)Math.pow(2, a) - 1;
        }
        return 1 + countNodes(root.left) + countNodes(root.right);
    }
}