剑指offer打卡|把二叉树打印成多行

题目描述

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

代码

import java.util.ArrayList;


/*
public class TreeNode {
    int val = 0;
    TreeNode left = null;
    TreeNode right = null;

    public TreeNode(int val) {
        this.val = val;

    }

}
*/
import java.util.*;
public class Solution {
    ArrayList<ArrayList<Integer> > Print(TreeNode pRoot) {
        ArrayList<ArrayList<Integer>> result=new ArrayList<>();
        if(pRoot==null){
            return result;
        }
        Queue<TreeNode> q=new LinkedList<>();
        ArrayList<Integer> list = new ArrayList<>();
        q.add(pRoot);
        int end=1;//当前层中还未打印的结点数
        int start=0;//当前层中已经打印过的结点数
        while(!q.isEmpty()){
            TreeNode node=q.poll();
            list.add(node.val);
            start++;
            if(node.left!=null){
                q.add(node.left);
            }
            if(node.right!=null){
                q.add(node.right);
            }
            //此时该打下一层了
            if(start==end){
                end=q.size();//此时end的值就是当前队列里的结点数，也就是待打印的结点数
                start=0;
                result.add(list);
                list=new ArrayList<Integer>();
            }
        }
        return result;
    }
    
}