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

小李子爱吃李子

于 2020-06-07 22:05:42 发布

阅读量87

点赞数

本文链接：https://blog.youkuaiyun.com/qq_34428410/article/details/106608497

版权

今日第二题：今天的题目是一道二叉树的题，它和“从上到下打印二叉树”十分类似，但是不需要多行，具体的解答见下面链接：
https://blog.youkuaiyun.com/qq_34428410/article/details/106453463
上题：

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

分析：需要从上到下、从左到右打印二叉树，我们如果做过上面这个题，知道要用队列实现，但是这里有增加了一个条件，按层输出一行，即还需要一个队列存储下面的节点。
步骤：
    1、建立两个Queue--one（存放当前结点）、two（存放孩子结点），用LinkedList,其实Queue的实现类。
    2、还需要建立一个列表arr，存放每一层的节点值。
    3、将当前要取值的节点存入one中；
    4、遍历one节点，取出当前结点，将该节点的值放入arr中；
    5、将one中当前结点的left和right放入two中；一致重复4-5，直到one中为空；
    6、将arr存入二维列表中，one和two进行交换，使得子孩子变为根；
代码：
import java.util.ArrayList;
import java.util.Queue;
import java.util.LinkedList;
/*
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) {
        Queue<TreeNode> one = new LinkedList<>();
        Queue<TreeNode> two = new LinkedList<>();
        one.offer(pRoot);
        ArrayList<ArrayList<Integer>> array = new ArrayList<ArrayList<Integer>>();
        if(pRoot==null)return array;
        while(!one.isEmpty()){
            ArrayList<Integer> arr = new ArrayList<>();
            while(!one.isEmpty()){
                TreeNode temp =one.poll();
                arr.add(temp.val);
                if(temp.left!=null){
                    two.offer(temp.left);
                }
                if(temp.right!=null){
                    two.offer(temp.right);
                }
            }
            array.add(arr);
            Queue<TreeNode> temp = one;
            one=two;
            two=temp;
        }
        return array;
    }
    
}
牛客运行通过
运行时间：29ms
运行内存：9600Kb
其实和链接中的题目很像，只是增加了一个存储子孩子的队列，但是思想还是一样的。
欢迎各位互相交流学习哦~