剑指offer_【60】把二叉树打印成多行

最新推荐文章于 2021-04-01 11:51:24 发布

原创最新推荐文章于 2021-04-01 11:51:24 发布 · 133 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#二叉树题

剑指offer 专栏收录该内容

65 篇文章

订阅专栏

本文介绍了一种从上到下按层打印二叉树的方法，通过使用队列辅助进行层次遍历，确保了同一层节点能从左至右输出。文章详细解释了解题思路，并提供了完整的Java代码实现。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

1.题目描述

从上到下按层打印二叉树，同一层结点从左至右输出。每一层输出一行。

2.解题思路

利用层次遍历二叉树的方式，用一个队列进行辅助，每次打印前取n = queue.size()，可以保证逐行打印。

3.代码

import java.util.*;

/*

public class TreeNode {
    int val = 0;
    TreeNode left = null;
    TreeNode right = null;
    public TreeNode(int val) {

        this.val = val;

    }
}

*/
public class Solution {
   ArrayList<ArrayList<Integer>> Print(TreeNode pRoot) {
       ArrayList<ArrayList<Integer>> res = new ArrayList<>();
       if(pRoot == null) {
           return res;
       }
       Queue<TreeNode> queue = new LinkedList<>();
       //加入根结点
       queue.add(pRoot);
       while (!queue.isEmpty()){
           //每遍历一次，就新建一次tmp
           ArrayList<Integer>  tmp = new ArrayList<>();
           int count = queue.size();
           for(int i = 0;i<count;i++){
               if(queue.peek().left!=null){
                   //加入左结点
                   queue.add(queue.peek().left);
               }
               if(queue.peek().right!=null){
                   //加入右结点
                   queue.add(queue.peek().right);
               }
               //删除并返回queue中的头元素，删掉根（上一层结点元素）
               tmp.add(queue.poll().val);
           }
           //将一层的左右结点加入res
           res.add(tmp);
       }
       return res;
    }
}