剑指Offer——把二叉树打印成多行

Posted 2022-04-12 baoziy

1、题目描述

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

2、代码实现

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

/*
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>> listAll = new ArrayList<>();
        ArrayList<Integer> list = new ArrayList<>();
        Queue<TreeNode> queue = new LinkedList<>();
        if (pRoot == null) 
            return listAll;
        
        int start = 0;
        int end = 1;
        queue.add(pRoot);
        while (!queue.isEmpty()) 
            TreeNode temp = queue.remove();
            list.add(temp.val);
            start++;
            if (temp.left != null) 
                queue.add(temp.left);
            
            if (temp.right != null) 
                queue.add(temp.right);
            
            //这里的end记录的是已经存入queue中的下一层的节点个数
            //当start等于end的事后证明该层的节点已经全部存入了list链表中，需要把list作为对象存入list中
            if (start == end) 
                //此时把下一层的节点个数也就是队列的长度重新作为end值
                end = queue.size();
                //start需要重新置为0
                start = 0;
                listAll.add(list);
                //为了避免上一组的list对后面的操作造成影响，重新创建一个ArrayList<>()赋值给list
                list = new ArrayList<>();
            
        
        return listAll;