已知一棵完全二叉树,求其节点个数。
要求:时间复杂度低于 O(N),N为这棵树的节点个数
已知,如果一棵满二叉树的高度为l,则它的节点个数为2^l-1
如下的一棵完全二叉树
因为是完全二叉树,所以只要通过遍历它的左边界,就能知道它的层数H,代价为O(logN)
然后遍历头节点右子树的左边界,如果右子树的左边界到了最后一层,说明左子树是满的,所以左树的节点数就能确定,是2^(H-1)-1
,再加上头节点,就是2^(H-1)
个,剩下右树也是完全二叉树,递归,求右树的节点数。
另一种情况,右子树的左边界没到最后一层,说明右子树是满的,只不过高度是H -2,所以左子树加上头节点,就有2^(H-2)个节点,接下来,左树又是一棵完全二叉树,递归。
class CompleteTreeNodeNumber{
public:
class Node{
public:
int value;
Node *left;
Node *right;
Node(int data){
this->value=data;
}
};
int nodeNum(Node *head){
if(head==nullptr){
return 0;
}
return bs(head,1,mostLeftLevel(head,1));
}
//node: 当前节点,level:当前节点的层数,h:整棵树的深度
int bs(Node *node,int level,int h){
if (level==h){
return 1;
}
//看右子树最左的节点到了哪一层,如果和h相同,
if(mostLeftLevel(node->right,level+1)==h){
//就返回左树的节点总数加头节点个数,加遍历右树
return((1<<(h-level))+bs(node->left,level+1,h));
}
else{
return((1<<(h-level-1))+bs(node->left,level+1,h));
}
}
int mostLeftLevel(Node*node, int level){
while (node!= nullptr){
level++;
node=node->left;
}
return level-1;
}
};