剑指offer: 按层进行打印二叉树

题目叙述:

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

题目分析:

首先是使用队列对二叉数进行层序遍历,但是关键点在于能够利用队列的动态变化的长度得到目前应该打印的层。

代码如下:

public class Solution {
    ArrayList<ArrayList<Integer> > Print(TreeNode pRoot) {
        ArrayList<ArrayList<Integer>> res = new ArrayList<>();
        if (pRoot == null) return res;
        //层序便利二叉树
        //声明一个队列
        Queue<TreeNode> queue = new LinkedList<>();//声明队列
        ArrayList<Integer> lis = new ArrayList<>();
        queue.offer(pRoot);
        int start =0, end =1;
        while(queue.peek()!=null){//或者!layer.isEmpty())
            //取出队首的元素;
            TreeNode top = queue.remove();
            lis.add(top.val);
            start++;
            if(top.left!=null) queue.offer(top.left);
            if(top.right!=null) queue.offer(top.right);
            if(start==end){
                res.add(lis);
                end = queue.size();
                start = 0;
                lis = new ArrayList<Integer>();
            }
        }
        return res;   
    } 
}
发布了126 篇原创文章 · 获赞 5 · 访问量 1万+

猜你喜欢

转载自blog.csdn.net/zhpf225/article/details/103411779