JZ77 按之字形顺序打印二叉树

描述

给定一个二叉树,返回该二叉树的之字形层序遍历,(第一层从左向右,下一层从右向左,一直这样交替)

数据范围: 0≤n≤1500,树上每个节点的val满足 |val| <= 1500 
要求:空间复杂度:O(n) ,时间复杂度:O(n) 

例如:
给定的二叉树是{1,2,3,#,#,4,5}

JZ77 按之字形顺序打印二叉树_第1张图片


该二叉树之字形层序遍历的结果是

[

[1],

[3,2],

[4,5]

]

示例1

输入:

{1,2,3,#,#,4,5}

返回值:

[[1],[3,2],[4,5]]

说明:

如题面解释,第一层是根节点,从左到右打印结果,第二层从右到左,第三层从左到右。     

示例2

输入:

{8,6,10,5,7,9,11}

返回值:

[[8],[10,6],[5,7,9,11]]

示例3

输入:

{1,2,3,4,5}

返回值:

[[1],[3,2],[4,5]]

 

简单层序遍历,但是唯一难点在于层序遍历同时还要确定层次(曾经一度认为不在TreeNode里加个字段level就实现不了,直到考研做王道书才发现完全可以,且很简单,脑子太死了),其实每次处理主体转变为当前pop(poll)的节点即可,提前获取当前队列长度,while(size-->0)) 内层加一个size次的循环即可天然分层了

我的代码:

   public ArrayList > Print(TreeNode pRoot) {
        ArrayList > aList = new ArrayList >();
        Queue q=new LinkedList();
        int k=0;

        if(pRoot!=null) q.add(pRoot);
        while (!q.isEmpty()){
            int size=q.size();//确定层次 其实也很简单 当前队列里所有元素都是一层 处理完再处理下一层即可
            ArrayList list = new ArrayList();

            while (size-- >0){//一次性处理完一层
                TreeNode p = q.poll();
                list.add(p.val);//不仅逻辑统一了 代码也简洁不少
                if(p.left!=null)  q.add(p.left);
                if(p.right!=null) q.add(p.right);               
            }

            if((++k)%2==0) Collections.reverse(list);//根是第1层 然后奇数层要反转
            if(list.size()>0) aList.add(list);

        }
        return aList;
    }

你可能感兴趣的:(#,算法,数据结构)