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

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/count-complete-tree-nodes

---

题意：

给你一棵 完全二叉树 的根节点 root ，求出该树的节点个数。

完全二叉树 的定义如下：在完全二叉树中，除了最底层节点可能没填满外，其余每层节点数都达到最大值，并且最下面一层的节点都集中在该层最左边的若干位置。若最底层为第 h 层，则该层包含 1~ 2h 个节点。

示例 1：

输入：root = [1,2,3,4,5,6]
输出：6

示例 2：
输入：root = []
输出：0

示例 3：
输入：root = [1]
输出：1

提示：

树中节点的数目范围是[0, 5*10⁴]
0 <= Node.val <= 5 * 10⁴
题目数据保证输入的树是 完全二叉树

进阶：遍历树来统计节点是一种时间复杂度为 O(n) 的简单解决方案。你可以设计一个更快的算法吗？

---

参考文章

思路：

这道题我们先尝试用二叉树的遍历方法遍历每一个节点然后计数，分别使用递归法和迭代法。这种做法对所有二叉树都适用，并没有用到完全二叉树的特性，时间复杂度为O(N)，比较慢

递归法：

递归法的代码真的好简洁

Java代码：

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

迭代法：

迭代法这里仍然是借用二叉树层序遍历的模板，用一个变量计数节点数即可。二叉树层序遍历看我的这个文章 LeetCode上10道关于二叉树层序遍历的题目。

Java代码：

class Solution {
	public int countNodes(TreeNode root) {
		if (root == null)
			return 0;
		int ans = 0;
		Deque<TreeNode> deque = new LinkedList<>();
		deque.offer(root);
		while (!deque.isEmpty()) {
			int size = deque.size();
			for (int i = 0; i < size; i++) {
				TreeNode rnNode = deque.poll();
				ans++;
				if (rnNode.left != null)
					deque.offer(rnNode.left);
				if (rnNode.right != null)
					deque.offer(rnNode.right);
			}
		}
		return ans;
	}
}

利用完全二叉树的特性解题思路：

一棵完全二叉树要么是满二叉树，要么是最后一层叶子节点没有满。同时，完全二叉树根节点的左右孩子一定有一棵是满二叉树，另一颗也是完全二叉树或者空节点。

如果一棵完全二叉树是满二叉树，那么它的节点数是 2^树的深度 + 1。
如果一棵完全二叉树不是满二叉树，那么我们分别递归计算它的左孩子和右孩子的结点，它的节点数是 左孩子节点数 + 右孩子节点数 + 1。在这递归的过程中，递归到某一深度，一定会有左右孩子是满二叉树，就可以按照满二叉树的节点数计算公式来计算。

时间复杂度O(logN * logN)，比上面的递归法和迭代法更快

示例图（图片转自代码随想录）：

Java代码：

class Solution {
	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;
		else
			return countNodes(root.left) + countNodes(root.right) + 1;
	}
}