从上往下打印二叉树（java版）

最新推荐文章于 2023-03-09 21:42:46 发布

原创最新推荐文章于 2023-03-09 21:42:46 发布 · 899 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#java #剑指offer-java

算法-剑指Offer-java 专栏收录该内容

59 篇文章

订阅专栏

本文介绍了一种从上往下打印二叉树节点的方法，实现了通过层序遍历将同一层级的节点按从左至右的顺序输出。采用队列实现算法流程，先将根节点加入队列，然后循环取出队首节点并打印，再将其左右子节点依次加入队列，直至队列为空。

【题目描述】从上往下打印出二叉树的每个节点，同层节点从左至右打印。

【解题思路】
//1. 层序遍历二叉树。
//2. 使用一个ArrayList记录下依次访问到的节点。

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

public class Solution {
    public ArrayList<Integer> PrintFromTopToBottom(TreeNode root) {
        ArrayList<Integer> arr = new ArrayList<Integer>();
        Queue<TreeNode> q = new LinkedList<TreeNode>();
        if(root != null){
            q.offer(root);
        }
        while(!q.isEmpty()){
            TreeNode tr = q.poll();
            arr.add(tr.val);
            System.out.println(tr.val);
            if(tr.left != null){
                q.offer(tr.left);
            }
            if(tr.right != null){
                q.offer(tr.right);
            }
        }
        return arr;
    }
}