题解 | #把二叉树打印成多行#

把二叉树打印成多行

http://www.nowcoder.com/practice/445c44d982d04483b04a54f298796288

双队列实现,第一层打印完存第二层,第二层打印时存入第三层

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) {
        TreeNode root = pRoot; 
        Queue<TreeNode> queue = new LinkedList<>();
        Queue<TreeNode> queue2 = new LinkedList<>();
        ArrayList<ArrayList<Integer> > list = new ArrayList<>();
        TreeNode op;
        queue.offer(root);
        if(root == null){
            return list;
        }
        while(true){
            List<Integer> li = new ArrayList<>();

            while(!queue.isEmpty()){
                op = queue.poll();
                if(op.left != null){
                    queue2.offer(op.left);
                }
                if(op.right != null){
                    queue2.offer(op.right);
                }
                li.add(op.val);
            }
            if(li.size() != 0){
                list.add(new ArrayList(li));
            }
            li.clear();
            while(!queue2.isEmpty()){
                op = queue2.poll();
                if(op.left != null){
                    queue.offer(op.left);
                }
                if(op.right != null){
                    queue.offer(op.right);
                }
                li.add(op.val);
            } 
            if(li.size() != 0){
                list.add(new ArrayList(li));
            }
            li.clear();
            if(queue.isEmpty() && queue2.isEmpty()){
                break;
            }
        }
        return list;
    }
    
}
全部评论

相关推荐

仁者伍敌:服务员还要脱颖而出,这是五星级酒店吗
点赞 评论 收藏
分享
湫湫湫不会java:1.在校经历全删了2.。这些荣誉其实也没啥用只能说,要的是好的开发者不是好好学生3.项目五六点就行了,一个亮点一俩行,xxx技术解决,xxx问题带来xxx提升。第一页学历不行,然后啥有价值的信息也没有,到第二页看到项目了,第一个项目九点,第二个项目像凑数的俩点。总体给人又臭又长,一起加油吧兄弟
点赞 评论 收藏
分享
评论
点赞
收藏
分享

创作者周榜

更多
牛客网
牛客网在线编程
牛客网题解
牛客企业服务