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

把二叉树打印成多行

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

题目描述
从上到下按层打印二叉树,同一层结点从左至右输出。每一层输出一行。
1、思路分析
因为每层的结点均是从左往右打印,所以只需要用一个队列来进行存储即可,需要注意的是,题目要求每一层输出一行,为了区别不同的行数,需要记录当前层和下一层的结点数。采取添加-移除-添加的方式。
2、代码

public class Solution {
    ArrayList<ArrayList<Integer> > Print(TreeNode pRoot) {
        ArrayList<ArrayList<Integer>> res = new ArrayList<>();
        ArrayList<Integer> tmp = new ArrayList<>();
        LinkedList<TreeNode> tn = new LinkedList<>();
        if(pRoot == null) {
            return res;
        }
        tn.add(pRoot);
        int now = 1, next = 0;
        while(!tn.isEmpty()) {
            TreeNode t = tn.remove();
            now--;
            tmp.add(t.val);
            if(t.left != null) {
                tn.add(t.left);
                next++;
            }
            if(t.right != null) {
                tn.add(t.right);
                next++;
            }
            if(now == 0) {
                res.add(new ArrayList<Integer>(tmp));
                tmp.clear();
                now = next;
                next = 0;
            }
        }
        return res;
    }
}
全部评论

相关推荐

伟大的烤冷面被普调:暨大✌🏻就是强
点赞 评论 收藏
分享
孤寡孤寡的牛牛很热情:为什么我2本9硕投了很多,都是简历或者挂,难道那个恶心人的测评真的得认真做吗
点赞 评论 收藏
分享
评论
1
收藏
分享
牛客网
牛客企业服务