给定一棵完全二叉树的根节点root,返回这棵树的节点个数。如果完全二叉树的节点数为N,请实现时间复杂度低于O(N)的解法。给定树的根结点root,请返回树的大小。
二分的思想。看完全二叉树的最后的最右一个节点的位置
思路:
1 找到完全二叉树的最左节点,也就是求左子树的深度
2 找到完全二叉树头节点右子树中的最左节点,记录右子树深度
3 如果两个深度相等,说明头节点左子树是一棵满二叉树,使用公式求得长度再加上头节点,然后对于右子树使用递归求解
4 如果左子树深度大于右子树深度,说明右子树是一棵完全二叉树,使用公式求得长度再加上头节点,然后对于左子树使用递归求解.
public int count(TreeNode root) {
if(root==null) return 0;
int count = 0;
TreeNode cur = root.left;
//记录左子树的深度