乐趣区

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

齐全二叉树的节点个数

题目形容:给你一棵 齐全二叉树 的根节点 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));
    }
}

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

退出移动版