多种方法实现统计完全二叉树的节点个数

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

难度：中等

给你一棵 完全二叉树 的根节点 root ，求出该树的节点个数。

完全二叉树 的定义如下：在完全二叉树中，除了最底层节点可能没填满外，其余每层节点数都达到最大值，并且最下面一层的节点都集中在该层最左边的若干位置。若最底层为第 h 层，则该层包含 1~ 2^h 个节点。

思路

本文总共实现了三种方法

• 直接递归版本,没有利用到完全二叉树的特性
• 迭代版本,没有利用到完全二叉树的特性
• 递归版本,利用完全二叉树的特性进行计算

利用完全二叉树的特性进行计算的递归版本

对于完全二叉树中的一个节点，如果它的左子树的深度等于右子树的深度，那么该节点是一颗完全二叉树，可以使用公式$num=2^{\text{depth}}-1$直接计算该子树的节点个数，可以减少一部分的递归时间

/**
 * 递归版本,利用完全二叉树的特性进行计算
 *
 * @param root
 * @return
 */
public int countNodes(TreeNode root) {
    if (root == null) {
        return 0;
    }
    int leftDepth = 1;
    int rightDepth = 1;
    TreeNode leftNode = root.left;
    TreeNode rightNode = root.right;
    while (leftNode != null) {
        leftDepth++;
        leftNode = leftNode.left;
    }
    while (rightNode != null) {
        rightDepth++;
        rightNode = rightNode.right;
    }
    if (leftDepth == rightDepth) {
        return (int) Math.pow(2, leftDepth) - 1;
    }
    return countNodes(root.left) + countNodes(root.right) + 1;
}

直接递归版本

/**
 * 直接递归版本,没有利用到完全二叉树的特性
 *
 * @param root
 * @return
 */
public int countNodes(TreeNode root) {
    if (root == null) {
        return 0;
    }

    return countNodes(root.left) + countNodes(root.right) + 1;
}

迭代版本

/**
 * 迭代版本,没有利用到完全二叉树的特性
 *
 * @param root
 * @return
 */
public int countNodes(TreeNode root) {
    if (root == null) {
        return 0;
    }
    int nodeNum = 0;
    Deque<TreeNode> deque = new LinkedList<>();
    deque.addLast(root);
    while (!deque.isEmpty()) {
        nodeNum++;
        TreeNode node = deque.removeFirst();
        if (node.left != null) {
            deque.addLast(node.left);
        }
        if (node.right != null) {
            deque.addLast(node.right);
        }
    }

    return nodeNum;
}