题目(leecode T222):
给你一棵 完全二叉树 的根节点 root
,求出该树的节点个数。
完全二叉树 的定义如下:在完全二叉树中,除了最底层节点可能没填满外,其余每层节点数都达到最大值,并且最下面一层的节点都集中在该层最左边的若干位置。若最底层为第 h
层,则该层包含 1~ 2h
个节点。
方法:
层序遍历法:
完全二叉树指的是一棵深度为k的有n个结点的二叉树,对树中的结点按从上至下、从左到右的顺序进行编号,如果编号为i(1≤i≤n)的结点与满二叉树中编号为i的结点在二叉树中的位置相同,则这棵二叉树称为完全二叉树。要计算其中的节点的个数,使用层序遍历非常的容易,昨天的题目是计算深度,计算深度我们是在每一层的遍历结束之后给depth++,而计算节点的话我们就需要在处理每一个节点的时候就进行result++的操作。因此由层序遍历修改而来非常的容易。只需要在处理节点出队的语句后增加一个节点个数result++的语句即可。
class Solution {
public:
int countNodes(TreeNode* root) {
if(root == NULL) return 0;
queue<TreeNode*> que;
que.push(root);
int result = 0;
while(!que.empty()){
int size = que.size();
for(int i = 0; i<size; i++){
TreeNode* node = que.front();
que.pop();
result++; //在这里计算节点个数
if(node->left) que.push(node->left);
if(node->right) que.push(node->right);
}
}
return result;
}
};