关于java:LeetCode222完全二叉树的节点个数

9次阅读

共计 1209 个字符，预计需要花费 4 分钟才能阅读完成。

题目形容：给你一棵齐全二叉树的根节点 root，求出该树的节点个数。

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

示例阐明请见 LeetCode 官网。

起源：力扣（LeetCode）
链接：https://leetcode-cn.com/probl…
著作权归领扣网络所有。商业转载请分割官网受权，非商业转载请注明出处。

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

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

首先如果根节点为空，即为空树，阐明没有节点，间接返回 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));
    }
}