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

2022/4/15 23:43:30

本文主要是介绍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));
    }
}

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



这篇关于LeetCode-222-完全二叉树的节点个数的文章就介绍到这儿,希望我们推荐的文章对大家有所帮助,也希望大家多多支持为之网!


扫一扫关注最新编程教程