Leetcode: NO.103 二叉樹的鋸齒形層序遍歷 層序遍歷
阿新 • • 發佈:2020-12-23
技術標籤:演算法
題目
給定一個二叉樹,返回其節點值的鋸齒形層序遍歷。(即先從左往右,再從右往左進行下一層遍歷,以此類推,層與層之間交替進行)。
例如:
給定二叉樹 [3,9,20,null,null,15,7],
3
/ \
9 20
/ \
15 7
返回鋸齒形層序遍歷如下:
[
[3],
[20,9],
[15,7]
]
連結:https://leetcode-cn.com/problems/binary-tree-zigzag-level-order-traversal
解題記錄
- 層序遍歷,奇偶行寫入一個在後面加入,一個在前面
/**
* @author: ffzs
* @Date: 2020/12/22 下午7:54
*/
class TreeNode {
int val;
TreeNode left;
TreeNode right;
TreeNode(int x) { val = x; }
}
public class Solution {
public List<List<Integer>> zigzagLevelOrder(TreeNode root) {
List<List<Integer>> res = new ArrayList<>();
if (root == null) return res;
Deque<TreeNode> deque = new LinkedList<>();
int level = 0;
deque.add(root);
while (!deque.isEmpty()) {
int num = deque.size();
List<Integer> list = new ArrayList<>();
for (int i = 0; i < num; i++ ) {
TreeNode cur = deque.pop();
System.out.println(cur.val);
if ((level&1) == 0) list.add(cur.val);
else list.add(0, cur.val);
if (cur.left != null) deque.offer(cur.left);
if (cur.right != null) deque.offer(cur.right);
}
res.add(list);
level++;
}
return res;
}
}