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

题目描述

时间限制：1秒 空间限制：32768K

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

---

解题思路

前序遍历二叉树，对于每一层经过的结点总是从最左边的开始，然后才逐渐从左到右，其中可能经过不同层的结点，所以每一层做一下标记，不同层的结点添加到不同层的列表中。
例如

ArrayList<ArrayList<Integer>> ans;

那么，第d层的结点从左到右保存在ArrayList<Integer> arr = ans.get(d)中。

代码（Java）

import java.util.ArrayList;import java.util.Collections;class TreeNode {    int val = 0;    TreeNode left = null;    TreeNode right = null;    public TreeNode(int val) {        this.val = val;    }}public class Main {    public static ArrayList<ArrayList<Integer>> ans;    public static void dfs(TreeNode root, int d) {        //第d层还没有结点，创建存放第d层结点的数组列表        while(ans.size()-1 < d) {            ans.add(new ArrayList<Integer>());        }        //将该节点加入到对应的第d层的数组列表中        ans.get(d).add(root.val);        if(root.left != null)dfs(root.left, d+1);        if(root.right != null)dfs(root.right, d+1);    }    public static ArrayList<ArrayList<Integer>> Print(TreeNode pRoot) {        ans = new ArrayList<ArrayList<Integer>>();        if(pRoot == null) return ans;        ans.add(new ArrayList<Integer>());        dfs(pRoot, 0);        //for(int i = 1; i <= ans.size(); i++) {            //if((i&1) == 0) Collections.reverse(ans.get(i-1));        //}        return ans;    }    /*        测试代码     */    public static void main(String []args){        TreeNode root = new TreeNode(1);        root.left = new TreeNode(2);        root.left.left = new TreeNode(4);        root.left.right = new TreeNode(5);        root.right = new TreeNode(3);        root.right.left = new TreeNode(6);        root.right.right = new TreeNode(7);        ArrayList<ArrayList<Integer>> ans = Print(root);        for(int i = 0; i < ans.size(); i++) {            for(int a : ans.get(i))System.out.print(a+" ");            System.out.println();        }    }}