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

最新推荐文章于 2024-11-13 20:43:20 发布

贝耶儿

最新推荐文章于 2024-11-13 20:43:20 发布

阅读量418

点赞数 5

文章标签： java 算法数据结构

本文链接：https://blog.csdn.net/qq_44047415/article/details/135759514

版权

针对于完全二叉树的写法

public int countNodes(TreeNode root) {
        if (root == null) {
            return 0;
        }
        TreeNode left = root.left;
        TreeNode right = root.right;
        int leftDepth = 0, rightDepth = 0;
        while (left != null) {
            left = left.left;
            leftDepth++;
        }
        while (right != null) {
            right = right.right;
            rightDepth++;
        }
        if (leftDepth == rightDepth) {
            return (2 << leftDepth) - 1;
        }
        // 以root为根的树不是满二叉树
        return countNodes(root.left) + countNodes(root.right) + 1;
    }