力扣算法 Java 刷题笔记【二叉树篇】hot100（十一）如何计算完全二叉树的节点数及其时间复杂度分析 3

最新推荐文章于 2024-03-11 09:30:00 发布

心海非海_

最新推荐文章于 2024-03-11 09:30:00 发布

阅读量825

点赞数 2

分类专栏：数据结构与算法文章标签： leetcode java 算法二叉树

本文链接：https://blog.csdn.net/weixin_46644403/article/details/121716977

版权

数据结构与算法专栏收录该内容

51 篇文章 7 订阅

订阅专栏

文章目录

1. 普通二叉树节点个数
2. 满二叉树节点个数
3. 完全二叉树的节点个数

1. 普通二叉树节点个数

地址: https://labuladong.gitee.io/algo/2/18/31/
2021/12/15
做题反思：

int countNodes(TreeNode root){
	if (root == null) {
		return 0;
	}
	return countNodes(root.left) + countNodes(root.right) + 1;
}

时间复杂度 O(N)：

2. 满二叉树节点个数

地址: https://labuladong.gitee.io/algo/2/18/31/
2021/12/15
做题反思：

public int countNodes(TreeNode root) {
	int h = 0;
	while (root != null) {
		root = root.left;
		h++;
	}
	return (int)Math.pow(2, h) - 1;
}

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

地址: https://leetcode-cn.com/problems/count-complete-tree-nodes/
2021/12/15
做题反思：两个小问题

= 和 ==
if 和 while

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