剑指offer之把二叉树打印成多行（队列）

最新推荐文章于 2023-11-26 18:29:08 发布

原创最新推荐文章于 2023-11-26 18:29:08 发布 · 116 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#队列 #java #二叉树

剑指offer 专栏收录该内容

45 篇文章

订阅专栏

本文介绍了一种使用ArrayList实现二叉树节点层次遍历的方法，通过队列操作确保按层次顺序存储节点。核心思路是利用ArrayList存储每层节点，并将它们作为子列表添加到一个多层集合中。

思路：

ArrayList list存储每层结点，然后list作为单位元素存入ArrayList<ArrayList<>>中。

队列中每次循环存储下一层的所有结点。

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> > L = new ArrayList<ArrayList<Integer > >();
    if(pRoot == null)
        return L;
     Queue<TreeNode> que = new LinkedList <TreeNode>();
        //先进先出，将当前结点的左右孩子依次存入队列，队列中每次循环存储的
        //是下一层树的结点
     que.add(pRoot);
     while(!que.isEmpty()){
         ArrayList<Integer> list = new ArrayList <>();
         int len = que.size();//控制层数，每次打印一层
         for(int i = 0; i< len;i++){
             TreeNode temp = que.poll();
             list.add(temp.val);
             if(temp.left!=null)
                 que.add(temp.left);
             if(temp.right!= null)
                 que.add(temp.right);
         }
         L.add(list);//按层打印，一个list是一层
     }
        return L;
    }
}