剑指 Offer 32 - II. 从上到下打印二叉树 II

题目：

从上到下按层打印二叉树，同一层的节点按从左到右的顺序打印，每一层打印到一行。

 

例如:
给定二叉树: [3,9,20,null,null,15,7],

    3
   / \
  9  20
    /  \
   15   7

返回其层次遍历结果：

[
  [3],
  [9,20],
  [15,7]
]

 

提示：

1. 节点总数 <= 1000

代码： //超时

/**
 * Definition for a binary tree node.
 * public class TreeNode {
 *     int val;
 *     TreeNode left;
 *     TreeNode right;
 *     TreeNode(int x) { val = x; }
 * }
 */
class Solution {
    public List<List<Integer>> levelOrder(TreeNode root) {
        if(root==null){return null;}
        List<List<TreeNode>> list =new ArrayList<>();
        List<TreeNode> list1=new ArrayList<>();
        list1.add(root);
        list.add(list1);
        
        //一层一个动态数组，循环放入listz中
        for(int i=0;i<list.size();i++){
            List<TreeNode> tempList=list.get(i);
            List<TreeNode> list2=new ArrayList<>();
            for(int j=0;j<tempList.size();j++){
                TreeNode temp=tempList.get(i);
                if(temp.left!=null){list2.add(temp.left);}
                if(temp.right!=null){list2.add(temp.right);}
            }
            list.add(list2);
        }
        //转换成返回类型
        List<List<Integer>> res =new ArrayList<>();

        for(int i=0;i<list.size();i++){
            List<TreeNode> tempList=list.get(i);
            Iterator iterator =tempList.iterator();
            List<Integer> res1=new ArrayList<>();
            while(iterator.hasNext()){
                res1.add(((TreeNode)iterator.next()).val);
            }
            res.add(res1);

        }

        return res;
        
    }
}

代码2：

//大佬写的，我做不来，没想到可以用一个数t表示层数传入递归方法，然后在list中取出第t层的动态数组，然后直接放入动态数组中，膜拜大佬。

class Solution {
    List <List<Integer>> list=new ArrayList<>();
    public List<List<Integer>> levelOrder(TreeNode root) {
        fun(root,0);
        return list;

    }

    public void fun(TreeNode root,int t){
        if(root!=null){
            if(t>=list.size()){
                list.add(new ArrayList());
            }
            list.get(t).add(root.val);
            fun(root.left,t+1);
            fun(root.right,t+1);
        }
    }
}