题目描述:
给出一个完全二叉树,求出该树的节点个数。
说明:
完全二叉树的定义如下:在完全二叉树中,除了最底层节点可能没填满外,其余每层节点数都达到最大值,并且最下面一层的节点都集中在该层最左边的若干位置。若最底层为第 h 层,则该层包含 1~ 2h 个节点。
示例:
输入:
1
/ \
2 3
/ \ /
4 5 6
输出: 6
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/count-complete-tree-nodes
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
/**
* 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;
else return bs(root,1,mostLeftLevel(root,1));
}
private int bs(TreeNode node,int level,int h) {
if(level == h) return 1;
if(mostLeftLevel(node.right,level+1) == h){
return (1<<(h-level))+bs(node.right,level+1,h);
}else { // 1<< (h-level),代表2的h-level次方,代表 满二叉树的结点个数加上根节点
return (1<<(h - level -1))+bs(node.left,level+1,h);
}
}
private int mostLeftLevel(TreeNode node,int level) {
while (node != null){
level++;
node = node.left;
}
return level-1;
}
}
思路概述:
通过mostLeftLevel()方法,找出该结点的最左的子树的深度h
之后通过bs()方法,定义一个level来代表当前层数,若该结点的右子树的最左子树深度与其相同,则代表该节点的左子树为一颗满二叉树,可以通过高度求出结点数,易得高度为h-level。
如果不相同,则代表该结点的右子树为一颗满二叉树,可以通过高度求出结点数,高度为h-level-1。
每一次递归时记得level+1