描述
给定一个二叉树,返回该二叉树的之字形层序遍历,(第一层从左向右,下一层从右向左,一直这样交替)
数据范围: 0≤n≤1500,树上每个节点的val满足 |val| <= 1500
要求:空间复杂度:O(n) ,时间复杂度:O(n)
例如:
给定的二叉树是{1,2,3,#,#,4,5}
该二叉树之字形层序遍历的结果是
[
[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<ArrayList<Integer> > Print(TreeNode pRoot) {
ArrayList<ArrayList<Integer> > aList = new ArrayList<ArrayList<Integer> >();
Queue<TreeNode> q=new LinkedList<TreeNode>();
int k=0;
if(pRoot!=null) q.add(pRoot);
while (!q.isEmpty()){
int size=q.size();//确定层次 其实也很简单 当前队列里所有元素都是一层 处理完再处理下一层即可
ArrayList<Integer> list = new ArrayList<Integer>();
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;
}