题目链接:https://leetcode-cn.com/problems/count-complete-tree-nodes/
题目描述
给出一个完全二叉树,求出该树的节点个数。
说明:
完全二叉树的定义如下:在完全二叉树中,除了最底层节点可能没填满外,其余每层节点数都达到最大值,并且最下面一层的节点都集中在该层最左边的若干位置。若最底层为第 h 层,则该层包含 1~ 2h 个节点。
示例:
输入:
1
/ \
2 3
/ \ /
4 5 6
输出: 6
思路
1 层序遍历
/*
* 层序遍历记录节点数
*/
class Solution {
public:
int countNodes(TreeNode* root) {
if(!root) return 0;
queue<TreeNode*> q;
int ret = 0;
q.push(root);
while (!q.empty()){
TreeNode *node = q.front();
++ret;
q.pop();
if(node->left)
q.push(node->left);
if(node->right)
q.push(node->right);
}
return ret;
}
};
2 递归
/*
* 递归
*/
class Solution {
public:
int countNodes(TreeNode* root) {
if(!root) return 0;
int leftNodes = countNodes(root->left);
int rightNodes = countNodes(root->right);
return leftNodes + rightNodes + 1;
}
};
3 递归+完全二叉树性质 O(log n)
前面两种方法都是暴力方法,时间复杂度为O(n),没有利用到完全二叉树的性质。
注意:完全二叉树的子树也肯定是完全二叉树
完全二叉树的节点数可以直接计算,所以我们采用二分查找的方差,递归地去计算不是完全二叉树的子树节点数。
二分搜索+递归:
(1)先判断左子树是不是完全二叉树,如果是的话通过树的高度可以计算得到左子树的节点数,总节点数=左子树节点数+1+右子树节点数(递归计算)
(2)如果不是,则右子树必然是完全二叉树,通过高度可以得到右子树的节点数,总节点数=右子树节点数+1+左子树节点数(递归计算)
复杂度分析
- 时间复杂度:O(log n)
/*
* 递归+完全二叉树性质
* O(log N)
*/
class Solution {
public:
int countNodes(TreeNode* root) {
if(!root) return 0;
int h = height(root);
if(h == height(root->right) + 1) // 右子树高度=根高度-1,左子树一定是完全二叉树
return (1<< (h-1)) + countNodes(root->right);
else // 右子树高度=根高度-2
return (1 << (h-2)) + countNodes(root->left);;
}
private:
// 完全二叉树高度
int height(TreeNode *root){
int count = 0;
while (root){
root = root->left;
count ++;
}
return count;
}
};