力扣面试150 完全二叉树的节点个数 树的高度

Problem: 222. 完全二叉树的节点个数

👨‍🏫 参考题解

Java 复制代码
/**
 * 二叉树节点的定义。
 * public class TreeNode {
 *     int val; // 节点的值
 *     TreeNode left; // 左子节点
 *     TreeNode right; // 右子节点
 *     TreeNode(int x) { val = x; } // 构造函数初始化节点值
 * }
 */

class Solution {
    // 计算完全二叉树的节点总数
    public int countNodes(TreeNode root) {
        // 如果根节点为null,说明这棵树是空树,节点数为0
        if (root == null) {
            return 0;
        }
        
        // 计算左子树的高度
        int left = countLevel(root.left);
        // 计算右子树的高度
        int right = countLevel(root.right);
        
        // 如果左子树和右子树的高度相等,说明左子树是一个满二叉树
        if (left == right) {
            // 左子树是满二叉树,其节点数为 2^left - 1,加上根节点则为 2^left
            // 然后递归计算右子树的节点总数
            return countNodes(root.right) + (1 << left); // 1<<left 相当于 2^left
        } else {
            // 如果左右子树的高度不相等,说明右子树是一个满二叉树
            // 右子树的节点数为 2^right - 1,加上根节点为 2^right
            // 然后递归计算左子树的节点总数
            return countNodes(root.left) + (1 << right); // 1<<right 相当于 2^right
        }
    }
    
    // 计算树的高度,只需沿着左子树一直向下遍历
    private int countLevel(TreeNode root) {
        int level = 0;
        // 不断遍历左子树,直到遍历到null为止,记录遍历的层数
        while (root != null) {
            level++; // 每向下遍历一层,层数加1
            root = root.left; // 沿着左子树继续向下遍历
        }
        // 返回树的高度
        return level;
    }
}
相关推荐
许愿与你永世安宁5 分钟前
力扣343 整数拆分
数据结构·算法·leetcode
爱coding的橙子8 分钟前
每日算法刷题Day42 7.5:leetcode前缀和3道题,用时2h
算法·leetcode·职场和发展
满分观察网友z39 分钟前
从一次手滑,我洞悉了用户输入的所有可能性(3330. 找到初始输入字符串 I)
算法
YuTaoShao1 小时前
【LeetCode 热题 100】73. 矩阵置零——(解法二)空间复杂度 O(1)
java·算法·leetcode·矩阵
Heartoxx1 小时前
c语言-指针(数组)练习2
c语言·数据结构·算法
大熊背1 小时前
图像处理专业书籍以及网络资源总结
人工智能·算法·microsoft
满分观察网友z1 小时前
别怕树!一层一层剥开它的心:用BFS/DFS优雅计算层平均值(637. 二叉树的层平均值)
算法
杰克尼2 小时前
1. 两数之和 (leetcode)
数据结构·算法·leetcode
YuTaoShao3 小时前
【LeetCode 热题 100】56. 合并区间——排序+遍历
java·算法·leetcode·职场和发展
二进制person7 小时前
Java SE--方法的使用
java·开发语言·算法