給定一個二叉樹,返回其節點值的鋸齒形層次遍歷。(即先從左往右,再從右往左進行下一層遍歷,以此類推,層與層之間交替進行)。node
例如:bash
給定二叉樹 [3,9,20,null,null,15,7],ui
3
/ \
9 20
/ \
15 7
複製代碼
返回鋸齒形層次遍歷以下:spa
[
[3],
[20,9],
[15,7]
]
複製代碼
/**
* Definition for a binary tree node.
* public class TreeNode {
* int val;
* TreeNode left;
* TreeNode right;
* TreeNode(int x) { val = x; }
* }
*/
class Solution {
public List<List<Integer>> zigzagLevelOrder(TreeNode root) {
List<List<Integer>> res = new ArrayList<List<Integer>>();
if( root == null ) return res;
Deque<TreeNode> tree = new LinkedList<>();
tree.add( root );
TreeNode end = root;
Boolean flag = true;
List<Integer> list = new ArrayList<>();
while( !tree.isEmpty() ){
TreeNode cur = flag ? tree.pollFirst() : tree.pollLast();
list.add( cur.val );
if( flag ){
if( cur.left != null ){
tree.offerLast( cur.left );
}
if( cur.right != null ){
tree.offerLast( cur.right );
}
}else{
if( cur.right != null ){
tree.offerFirst( cur.right );
}
if( cur.left != null ){
tree.offerFirst( cur.left );
}
}
if( cur == end ){
res.add( list );
list = new ArrayList<>();
end = flag ? tree.peekFirst() : tree.peekLast();
flag = !flag;
}
}
return res;
}
}
複製代碼
解題思路: 使用雙向鏈表保存下一層的值 , 用end指向當前層遍歷的最後一個結點,若是當前層的遍歷順序是從左到右 , 則每次彈出的都是鏈表的第一個元素 , 而後再把左右結點壓入鏈表的尾部,先壓左再壓右;若是當前層的遍歷順序是從右到左,則每次彈出的都是最後一個結點 , 而後把右左壓入到頭部,先壓右再壓左。code