LeetCode剑指 Offer 32 - III. 从上到下打印二叉树 III

LeetCode剑指 Offer 32 - III. 从上到下打印二叉树 III_第1张图片

 这道题比较中规中矩,就是一个BFS,不过有一个每一层输出顺序的问题(第一层从左到右打印,第二层从右到左打印)

看代码:

public static List> levelOrder(TreeNode root) {
        List> list = new ArrayList<>();
        if (root == null){//根节点空,返回空的list就好了
            return list;
        }
        Queue queue = new LinkedList<>();//存放每一层节点元素的队列
        queue.offer(root);
        while (!queue.isEmpty()){
            LinkedList res = new LinkedList<>();//存放每一层的打印值
            int size = queue.size();
            for (int i = 0; i < size; i++) {
                TreeNode node = queue.poll();
                if(list.size() % 2 ==0){//层数是偶数——>从右向左打印
                    res.addLast(node.data);
                }else {//层数是奇数——>从右向左打印
                    res.addFirst(node.data);
                }
                //添加下一层节点
                if(node.left!=null){
                    queue.offer(node.left);
                }
                if(node.right!=null){
                    queue.offer(node.right);
                }
            }
            list.add(res);
        }
        return list;
    }

可以看到存储每一层的打印值的时候是用到了一个LinkedList链表,我们知道LinkedList是一个双向链表,对于头尾节点操作很方便,所以如果层数是偶数,需要从右向左打印我们就将左边的节点保存在链表末尾,这样就实现了从右向左打印

你可能感兴趣的:(leetCode算法题,leetcode,算法)