0
点赞
收藏
分享

微信扫一扫

JAVA练习69-I. 从上到下打印二叉树

金牛豆豆 2022-01-31 阅读 59

从上到下打印出二叉树的每个节点,同一层的节点按照从左到右的顺序打印。

例如:
给定二叉树: [3,9,20,null,null,15,7],

        3
       /  \
     9   20
    /  \
 15   7

返回:
[3,9,20,15,7]

提示:

  • 节点总数 <= 1000

分析:

方法:队列

经典的广度优先(BFS)遍历题,这道题的解题思路在于每层按循序依次遍历,所以我们可以用队列存储每一层的节点,遍历完一个节点就将该节点的左右子节点入栈,这样这层遍历完后,队列里的节点就是下一层的节点,然后开始下一层遍历,如此重复,直到遍历完成。

时间复杂度:O(n) 
空间复杂度:O(n)

/**
 * Definition for a binary tree node.
 * public class TreeNode {
 *     int val;
 *     TreeNode left;
 *     TreeNode right;
 *     TreeNode(int x) { val = x; }
 * }
 */
class Solution {
    public int[] levelOrder(TreeNode root) {
        //判断空树
        if(root == null){
            return new int[]{};
        }
        //定义队列
        Deque<TreeNode> queue = new ArrayDeque<>();
        //入栈
        queue.offerLast(root);
        //创建List集合存储节点值
        List<Integer> list = new ArrayList<>();
        //遍历
        while(!queue.isEmpty()){
            //出栈
            TreeNode node = queue.pollFirst();
            //添加节点值
            list.add(node.val);
            //左节点不为空添加左节点
            if(node.left != null){
                queue.offerLast(node.left);
            }
            //右节点不为空添加右节点
            if(node.right != null){
                queue.offerLast(node.right);
            }
        }
        //获取长度
        int len = list.size();
        //创建数组
        int[] res = new int[len];
        //数组赋值
        for(int i = 0; i < len; ++i){
            res[i] = list.get(i);
        }
        return res;
    }
}

题目来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/cong-shang-dao-xia-da-yin-er-cha-shu-lcof

举报

相关推荐

0 条评论