给出一个完全二叉树,求出该树的节点个数。
说明:
完全二叉树的定义如下:在完全二叉树中,除了最底层节点可能没填满外,其余每层节点数都达到最大值,并且最下面一层的节点都集中在该层最左边的若干位置。若最底层为第 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 root) {
if(root ==null){
return 0;
}else{
return countNodes(root.left)+countNodes(root.right)+1;
}
}
}
这种方法运行遇到数据量多会超时失败
方法二:
思路:
1)根节点的右子树高度等于根节点左子树的高度时,说明根节点的左子树是满二叉树,可以使用公式 2^h-1。右子树可以使用递归的方式计算。
2)根节点的右子树高度不等于根节点左子树的高度时,说明根节点的右子树是满二叉树。而左子树,应该使用递归的方式解决。
/**
* 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 count(root);
}
}
public int count(TreeNode root){
if(root == null){
return 0;
}
//根节点左子树总高度
int lh=countHeight(root.left);
//计算根节点右子树高度
int rh=countHeight(root.right);
if(lh == rh){
//根节点左子树是满二叉树,遍历右子树遍历
return (lh>0? 1<<lh:1) + count(root.right);//1<<3 等于1*2^3
}else{
//根节点的右子树是满二叉树,遍历左子树
return (rh>0?1<<rh:1) + count(root.left);
}
}
//计算高度
public int countHeight(TreeNode node){
if(node == null){
return 0;
}else{
return countHeight(node.left)+1;
}
}
}
参考链接