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

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

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

示例 1：

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

示例 2：

输入：root = []
输出：0

示例 3：

输入：root = [1]
输出：1

提示：

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

来源：力扣（LeetCode）
链接：https://leetcode.cn/problems/count-complete-tree-nodes
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

感谢小成同学的详细解法完全二叉树的节点个数（两种方法：朴素遍历、利用完全二叉树的性质优化）

class Solution {
    public int countNodes(TreeNode root) {
//		return countNodesI(root);
		return countNodesII(root);
	}

	//方法二：利用完全二叉树特性，时间复杂度O(log²(n))
	private int countNodesII(TreeNode root) {
		if (root == null) {
			return 0;
		}
		int lh = getHeight(root.left);
		int rh = getHeight(root.right);
		//左子树必满 Math.pow(2, lh) - 1 + 右树节点数 + 1
		if (lh == rh) {
			return (1 << lh) + countNodesII(root.right);
		} else {
			//右子树必满 Math.pow(2, rh) - 1 + 左树节点数 + 1
			return (1 << rh) + countNodesII(root.left);
		}
	}

	private int getHeight(TreeNode root) {
		int height = 0;
		//因为完全二叉树是基于左节点的，所以我们可以遍历到最深的左节点，此时的height就是树高
		while (root != null) {
			root = root.left;
			height++;
		}
		return height;
	}

	//方法一：递归，时间复杂度O(log²(n))
	private int countNodesI(TreeNode root) {
		if (root == null) {
			return 0;
		}
		int lh = 0, rh = 0;
		TreeNode left = root, right = root;

		while (left != null) {
			lh++;
			left = left.left;
		}
		while (right != null) {
			rh++;
			right = right.right;
		}
		//满二叉树
		if (lh == rh) {
			return (1 << lh) - 1;
		}
		return countNodes(root.left) + countNodes(root.right) + 1;
	}
}