给出一个完全二叉树,求出该树的节点个数。
说明:
完全二叉树的定义如下:在完全二叉树中,除了最底层节点可能没填满外,其余每层节点数都达到最大值,并且最下面一层的节点都集中在该层最左边的若干位置。若最底层为第 h 层,则该层包含 1~ 2h 个节点。
示例:
输入: 1 / \ 2 3 / \ / 4 5 6 输出: 6
这篇博客说的很详细,大家可以看一下
https://blog.csdn.net/jmspan/article/details/51056085
/**
* Definition for a binary tree node.
* public class TreeNode {
* int val;
* TreeNode left;
* TreeNode right;
* TreeNode(int x) { val = x; }
* }
*/
class Solution {
public int countNodes(TreeNode root) {
if(root==null) return 0;
int count=1;
int left=leftMax(root.left);
int middle=leftMax(root.right);
if(left==middle){
count+=(1<<left)-1;
count+=countNodes(root.right);
}
else{
count+=(1<<middle)-1;
count+=countNodes(root.left);
}
return count;
}
public int leftMax(TreeNode root){
if(root==null) return 0;
int count=0;
while(root!=null){
count++;
root=root.left;
}
return count;
}
}