《剑指offer》把二叉树打印成多行（Java）

最新推荐文章于 2021-12-19 00:12:24 发布

原创最新推荐文章于 2021-12-19 00:12:24 发布 · 137 阅读

0 ·

CC 4.0 BY-SA版权

剑指offer 专栏收录该内容

63 篇文章

订阅专栏

本文介绍了一种二叉树的层次遍历算法，通过使用队列实现从上到下，同一层从左至右的节点输出。文章提供了一个完整的Java代码示例，展示了如何利用队列和ArrayList来实现这一功能。

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

题目描述

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

AC代码

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 root) {
        ArrayList<ArrayList<Integer>> res=new ArrayList<ArrayList<Integer>>();
        Queue<TreeNode> q=new LinkedList<>();
        ArrayList<Integer> level=new ArrayList<Integer>();
        q.add(root);
        q.add(null);
        while(q.size()!=0){
            TreeNode t=q.peek();
            q.poll();
            if(t==null){
                if(level.size()==0)
                    break;
                ArrayList<Integer> list=new ArrayList<>(level);
                res.add(list);
                q.add(null);
                level.clear();
                continue;
            }
            level.add(t.val);
            if(t.left!=null)
                q.add(t.left);
            if(t.right!=null)
                q.add(t.right);
                
        }
        return res;
    }
     
}