请实现一个函数按照之字形打印二叉树,即第一行按照从左到右的顺序打印,
第二层按照从右至左的顺序打印,第三行按照从左到右的顺序打印,其他行以此类推。
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
|
/** *
大家的实现很多都是将每层的数据存进ArrayList中,偶数层时进行reverse操作, *
在海量数据时,这样效率太低了。 *
(我有一次面试,算法考的就是之字形打印二叉树,用了reverse, *
直接被鄙视了,面试官说海量数据时效率根本就不行。) * *
下面的实现:不必将每层的数据存进ArrayList中,偶数层时进行reverse操作,直接按打印顺序存入 *
思路:利用Java中的LinkedList的底层实现是双向链表的特点。 *
1)可用做队列,实现树的层次遍历 *
2)可双向遍历,奇数层时从前向后遍历,偶数层时从后向前遍历 */ public
ArrayList<ArrayList<Integer> > Print(TreeNode pRoot) { ArrayList<ArrayList<Integer>>
ret = new
ArrayList<>(); if
(pRoot == null )
{ return
ret; } ArrayList<Integer>
list = new
ArrayList<>(); LinkedList<TreeNode>
queue = new
LinkedList<>(); queue.addLast( null ); //层分隔符 queue.addLast(pRoot); boolean
leftToRight = true ; while
(queue.size() != 1 )
{ TreeNode
node = queue.removeFirst(); if
(node == null )
{ //到达层分隔符 Iterator<TreeNode>
iter = null ; if
(leftToRight) { iter
= queue.iterator(); //从前往后遍历 } else
{ iter
= queue.descendingIterator(); //从后往前遍历 } leftToRight
= !leftToRight; while
(iter.hasNext()) { TreeNode
temp = (TreeNode)iter.next(); list.add(temp.val); } ret.add( new
ArrayList<Integer>(list)); list.clear(); queue.addLast( null ); //添加层分隔符 continue ; //一定要continue } if
(node.left != null )
{ queue.addLast(node.left); } if
(node.right != null )
{ queue.addLast(node.right); } } return
ret; } |