leetcode 222题 完全二叉树的节点个数
给你一棵 完全二叉树 的根节点 root ，求出该树的节点个数。
完全二叉树 的定义如下：在完全二叉树中，除了最底层节点可能没填满外，其余每层节点数都达到最大值，并且最下面一层的节点都集中在该层最左边的若干位置。若最底层为第 h 层，则该层包含 1~ 2h 个节点。
示例 1：
输入：root = [1,2,3,4,5,6]
输出：6
示例 2：
输入：root = []
输出：0
示例 3：
输入：root = [1]
输出：1
思路
bfs 层序遍历，累计
代码
/**
* 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) {
if (root === null) {
return 0;
}
const queue = [root];
let res = 0;
while (queue.length) {
const node = queue.shift();
res++;
if (node.left) {
queue.push(node.left);
}
if (node.right) {
queue.push(node.right);
}
}
return res;
};
function countNodes(root) {
if (root === null) {
return 0;
}
const queue = [root];
let curNum = 1, nextLevelNum = 0, res = 1;
while (queue.length) {
if (curNum > 0) {
const node = queue.shift();
if (node.left) {
queue.push(node.left);
nextLevelNum++;
}
if (node.right) {
queue.push(node.right);
nextLevelNum++;
}
curNum--;
}
if (curNum === 0) {
res += nextLevelNum;
curNum = nextLevelNum;
nextLevelNum = 0;
}
}
return res;
}
复杂度分析
- 时间复杂度：O(N)
- 空间复杂度：O(N)