Appearance
222. 完全二叉树的节点个数
题目描述
给你一棵** 完全二叉树** 的根节点 root ,求出该树的节点个数。
完全二叉树 的定义如下:在完全二叉树中,除了最底层节点可能没填满外,其余每层节点数都达到最大值,并且最下面一层的节点都集中在该层最左边的若干位置。若最底层为第 h 层(从第 0 层开始),则该层包含 1~ 2h 个节点。
示例 1:
**输入:**root = [1,2,3,4,5,6] **输出:**6
示例 2:
**输入:**root = [] **输出:**0
示例 3:
**输入:**root = [1] **输出:**1
提示:
树中节点的数目范围是[0, 5 * 104]
0 <= Node.val <= 5 * 104
- 题目数据保证输入的树是 **完全二叉树**
**进阶:**遍历树来统计节点是一种时间复杂度为 O(n) 的简单解决方案。你可以设计一个更快的算法吗?
难度: Easy
题解代码
javascript
/**
* Definition for a binary tree node.
* function TreeNode(val, left, right) {
* this.val = (val===undefined ? 0 : val)
* this.left = (left===undefined ? null : left)
* this.right = (right===undefined ? null : right)
* }
*/
/**
* @param {TreeNode} root
* @return {number}
*/
var countNodes = function(root) {
const queue = []
queue.push(root)
return bfs(root, queue, 0)
};
function bfs (root, queue, count) {
if (!root) return count
while(queue.length) {
const cur = queue.shift()
count++
if (cur.left) queue.push(cur.left)
if (cur.right) queue.push(cur.right)
}
return count
}