目录
二叉树的层次遍历
给你一个二叉树,请你返回其按 层序遍历 得到的节点值。 (即逐层地,从左到右访问所有节点)。
/**
* 102. 二叉树的层序遍历
* 给你一个二叉树,请你返回其按 层序遍历 得到的节点值。 (即逐层地,从左到右访问所有节点)。
*
*
*
* 示例:
* 二叉树:[3,9,20,null,null,15,7],
*
* 3
* / \
* 9 20
* / \
* 15 7
* 返回其层序遍历结果:
*
* [
* [3],
* [9,20],
* [15,7]
* ]
* https://leetcode-cn.com/problems/binary-tree-level-order-traversal/
*/
public List<List<Integer>> levelOrder(TreeNode root) {
List<List<Integer>> result = new ArrayList<>();
if (root == null) return result;
Queue<TreeNode> queue = new LinkedList<>();
queue.add(root);
while (!queue.isEmpty()) {
int size = queue.size();
List<Integer> oneLayerList = new ArrayList<>();// 每一层所有元素
while (size > 0) {
size--;
TreeNode temp = queue.poll();
if (temp != null) {
oneLayerList.add(temp.val);
if (temp.left != null) queue.add(temp.left);//queue是队列先进先出
if (temp.right != null) queue.add(temp.right);
}
}
result.add(oneLayerList);
}
return result;
}
二叉树的锯齿形层次遍历
给定一个二叉树,返回其节点值的锯齿形层序遍历。(即先从左往右,再从右往左进行下一层遍历,以此类推,层与层之间交替进行)。
例如:
给定二叉树 [3,9,20,null,null,15,7],
3
/ \
9 20
/ \
15 7
返回锯齿形层序遍历如下:
[
[3],
[20,9],
[15,7]
]
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/binary-tree-zigzag-level-order-traversal
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
/**
* 广度优先(BFS)
*/
public List<List<Integer>> zigzagLevelOrder(TreeNode root) {
List<List<Integer>> result = new ArrayList<>();
if (root == null) return result;
Queue<TreeNode> queue = new LinkedList<>();//正向顺序记录
Deque<Integer> stack = new LinkedList<>();// 反向顺序记录(想比较按照层打印多了该stack)
queue.add(root);
int layer = 1;// 丛1层开始
while (!queue.isEmpty()) {
int size = queue.size();
List<Integer> oneLayerList = new ArrayList<>();
while (size > 0) {
size--;
TreeNode temp = queue.poll();
if (temp != null) {
if (layer % 2 == 0) {// 偶数层用反向
stack.push(temp.val);
} else {// 奇数层正向
oneLayerList.add(temp.val);
}
if (temp.left != null) queue.add(temp.left);
if (temp.right != null) queue.add(temp.right);
}
}
if (layer % 2 == 0) {
while (!stack.isEmpty()) {
oneLayerList.add(stack.pop());
}
}
result.add(oneLayerList);
layer++;
}
return result;
}