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

链接

牛客：把二叉树打印成多行
LeetCode：剑指 Offer 32 - II. 从上到下打印二叉树 II
102. 二叉树的层序遍历

题目描述

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

思路

和《剑指offer》系列 从上往下打印二叉树（Java）这题几乎一样，区别在于返回值需要额外处理下。

代码

牛客：

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>> ret = new ArrayList<>();
        Queue<TreeNode> queue = new LinkedList<>();
        queue.add(pRoot);
        while (!queue.isEmpty()) {
            ArrayList<Integer> list = new ArrayList<>();
            int cnt = queue.size();
            while (cnt-- > 0) {
                TreeNode node = queue.poll();
                if (node == null)
                    continue;
                list.add(node.val);
                queue.add(node.left);
                queue.add(node.right);
            }
            if (list.size() != 0)
                ret.add(list);
        }
        return ret;
    }
    
}

LeetCode：

class Solution {
    public List<List<Integer>> levelOrder(TreeNode root) {
        List<List<Integer>> list = new ArrayList<List<Integer>>();
        if (root == null)
            return list;
        Queue<TreeNode> queue = new LinkedList<>();
        queue.offer(root);
        while (!queue.isEmpty()) {
            int num = queue.size();
            List<Integer> list1 = new ArrayList<>();
            for(int i=0; i<num; i++){
                TreeNode treeNode = queue.poll();
                list1.add(treeNode.val);
                if (treeNode.left != null)
                    queue.offer(treeNode.left);
                if (treeNode.right != null)
                    queue.offer(treeNode.right);
            }
            list.add(list1);
        }
        return list;
    }
}