【leetcode】222.完全二叉树的节点个数

题目

leetcode原题链接

Q给你一棵 完全二叉树 的根节点 root ，求出该树的节点个数。
完全二叉树 的定义如下：在完全二叉树中，除了最底层节点可能没填满外，其余每层节点数都达到最大值，并且最下面一层的节点都集中在该层最左边的若干位置。若最底层为第 h 层，则该层包含 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
题目数据保证输入的树是 完全二叉树

思路

最简单的办法就是遍历整棵树，记录节点的数量，可以使用递归和迭代两种方法。

一、遍历递归法

• 递归函数的参数为树的节点，返回值为以传入节点为根节点的树的节点数量。
• 递归结束条件，当遇到空节点时，返回0，表示节点数量为0
• 单层递归逻辑：节点数量=左子树节点数量+右子树节点数量+1（自身）

二、迭代层序遍历

层序遍历利用队列实现，在遍历过程中记录节点数量即可。如果对层序遍历还不熟悉的话，可以看看这篇文章 【leetcode】102.二叉树的层序遍历

三、公式+递归

完全二叉树有两种情况，一是它是一棵满二叉树，这时候直接利用公式2^深度 - 1, 二是它的最后一层节点没有满，这时候可以递归它的左右子树，当遇到满二叉树时就利用公式计算。如下图中红色方框圈起来的就是满二叉树，可以利用公式计算。

• 递归函数参数为树的节点，返回值是以该节点为根节点的树的节点数量。
• 递归结束条件：当遇到空节点时返回0
• 单层递归逻辑：计算以当前传入节点为根节点的树，最左边节点的深度和最右边节点的深度，如果两者相等，说明是满二叉树，利用公式计算节点数量并返回。如果不相等，就利用前面普通递归的逻辑，即节点数量=左子树节点数量+右子树节点数量+1（自身）

代码

一、遍历递归法

var countNodes = function(root) {
    if(!root) return 0
    return countNodes(root.left) + countNodes(root.right) + 1
};

二、迭代层序遍历

var countNodes = function(root) {
    if(!root) return []
    let queue = [root]
    let count = 0
    while(queue.length){
        const len = queue.length
        const curNode = queue.shift()
        count++
        if(curNode.left) queue.push(curNode.left)
        if(curNode.right) queue.push(curNode.right)
    }

    return count
};

三、公式+递归

var countNodes = function(root) {
    if(!root) return 0

    let left = root.left
    let right = root.right
    let leftDepth = 1
    let rightDepth = 1

    while(left){
        left = left.left
        leftDepth++
    }

    while(right){
        right = right.right
        rightDepth++
    }

    if(leftDepth === rightDepth) return 2 ** leftDepth - 1
    return countNodes(root.left) + countNodes(root.right) + 1
};

关注我的专栏，每天更新三道leetcode题解，一起变强！