题目
请实现一个函数按照之字形顺序打印二叉树,即第一行按照从左到右的顺序打印,第二层按照从右到左的顺序打印,第三行再按照从左到右的顺序打印,其他行以此类推。
例如:
给定二叉树: [3,9,20,null,null,15,7],
3
/
9 20
/
15 7
返回其层次遍历结果:
[
[3],
[20,9],
[15,7]
]
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/cong-shang-dao-xia-da-yin-er-cha-shu-iii-lcof
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
方法1:BFS
1、Z字型的时候实现,可以通过step的奇偶来实现。
2、翻转可改进:
list 有翻转函数
if (step % 2 == 0) Collections.reverse(row);
Java实现
class Solution {
public List<List<Integer>> levelOrder(TreeNode root) {
List<List<Integer>> res = new ArrayList<>();
if (root == null) return res;
Queue<TreeNode> q = new LinkedList<>();
q.offer(root);
int step = 0;
while (!q.isEmpty()) {
int sz = q.size();
List<Integer> var = new ArrayList<>();
for (int i = 0; i < sz; i++) {
TreeNode cur = q.poll();
var.add(cur.val);
if (cur.left != null) q.offer(cur.left);
if (cur.right != null) q.offer(cur.right);
}
if (step % 2 != 0) {
Collections.reverse(var);
res.add(var);
} else {
res.add(var);
}
step++;
}
return res;
}
}