给出一个完全二叉树,求出该树的节点个数。
说明:
完全二叉树的定义如下:在完全二叉树中,除了最底层节点可能没填满外,其余每层节点数都达到最大值,并且最下面一层的节点都集中在该层最左边的若干位置。若最底层为第 h 层,则该层包含 1~ 2h 个节点。
示例:
输入:
1
/ \
2 3
/ \ /
4 5 6
输出: 6
思路:
/**
* 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 head) {
if (head == null)return 0;
return bs(head, 1, mostLeftLevel(head, 1));
}
//返回根为node,当前层数为l,总深度为h的结点个数
public int bs(TreeNode node, int l, int h) {
if (l == h) {
return 1;
}
if (mostLeftLevel(node.right, l + 1) == h) { //右子树最深一行最左为空
return (1 << (h - l)) + bs(node.right, l + 1, h); //右bs+左子树结点个数
} else { //右子树最深一行最左不为空
return (1 << (h - l - 1)) + bs(node.left, l + 1, h);//左bs+右子树结点个数
}
}
//计算树的高度
public int mostLeftLevel(TreeNode node, int level) {
while (node != null) {
level++;
node = node.left;
}
return level - 1;
}
}