LeetCode-222-完全二叉树的节点个数

完全二叉树的节点个数

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

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

示例说明请见LeetCode官网。

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/count-complete-tree-nodes/
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

解法一：二叉树的层序遍历

对二叉树层序遍历，按从上到下、从左到右的顺序遍历二叉树，其中利用了队列先进先出的特点。具体处理过程如下：

• 首先如果根节点为空，即为空树，说明没有节点，直接返回0；
• 否则，先将根节点添加到队列中，然后遍历队列中的节点，直到队列为空，遍历过程中记录节点的数目。

最后返回节点数。

import com.kaesar.leetcode.TreeNode;

import java.util.LinkedList;
import java.util.Queue;

public class LeetCode_222 {
    /**
     * 二叉树层序遍历
     *
     * @param root
     * @return
     */
    public static int countNodes(TreeNode root) {
        // 如果根节点为空，即为空树，说明没有节点，直接返回0
        if (root == null) {
            return 0;
        }
        // 利用队列先进先出的特性，对二叉树进行层序遍历
        Queue<TreeNode> nodes = new LinkedList<>();
        // 首先将根节点添加到队列中，作为第一层
        nodes.add(root);
        // 记录二叉树的节点数
        int count = 0;
        // 遍历队列中的节点，直到队列为空
        while (!nodes.isEmpty()) {
            count++;
            TreeNode cur = nodes.poll();
            if (cur.left != null) {
                nodes.add(cur.left);
            }
            if (cur.right != null) {
                nodes.add(cur.right);
            }
        }
        return count;
    }

    public static void main(String[] args) {
        /**
         * 测试用例：
         *       1
         *    2     3
         *   4 5   6
         */
        TreeNode root = new TreeNode(1);
        root.left = new TreeNode(2);
        root.right = new TreeNode(3);
        root.left.left = new TreeNode(4);
        root.left.right = new TreeNode(5);
        root.right.left = new TreeNode(6);

        // 期望输出： 6
        System.out.println(countNodes(root));
    }
}

【每日寄语】 大丈夫行事，当磊磊落落，如日月皎然。