剑指offer-按之字形顺序打印二叉树

题目描述

请实现一个函数按照之字形打印二叉树，即第一行按照从左到右的顺序打印，第二层按照从右至左的顺序打印，第三行按照从左到右的顺序打印，其他行以此类推。

 

使用双向链表，先把当前行按顺序打印，再取出当前行并将下一行存入链表

public ArrayList<ArrayList<Integer> > Print(TreeNode pRoot) {//树 双向链表 mytip
        ArrayList<ArrayList<Integer>> re = new ArrayList<>();
        LinkedList<TreeNode> queue = new LinkedList<>();
        if(null==pRoot){
            return re;
        }
        queue.add(null);
        queue.add(pRoot);
        
        boolean flag = true;
        while(queue.size()>1){//最后一个为null
            TreeNode node = queue.poll();//当前不为null时，取出并将孩子结点存入链表
            if(null==node){//当前为null是将链表中数打印
                ArrayList<Integer> list = new ArrayList<>();
                Iterator<TreeNode> iter = null;
                if(flag){
                    iter = queue.iterator();//正向
                }else{
                    iter = queue.descendingIterator();//反向
                }
                while(iter.hasNext()){
                    TreeNode n = iter.next();
                    list.add(n.val);
                }
                if(queue.size()>0){
                    queue.add(null);
                }          
                re.add(list);
                flag = !flag;
                continue;//必须加
            }
            if(null!=node.left){
                queue.add(node.left);
            }
            if(null!=node.right){
                queue.add(node.right);
            }
        }
        return re;
    }