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
题解
1、队列 + 迭代
/**
* 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>> ans = new ArrayList<List<Integer>>();
Queue<TreeNode> queue = new LinkedList<>();
if (root == null){
return ans;
}
queue.offer(root);
List<Integer> tmp = new ArrayList<>();
int nextLevel = 0;// 下一个层节点的数量
int curLevel = 1;// 当前节点的数量
while(!queue.isEmpty()) {
TreeNode node = queue.peek();
queue.poll();
tmp.add(node.val);
if (node.left != null) {
queue.offer(node.left);
nextLevel++;
}
if (node.right != null) {
queue.offer(node.right);
nextLevel++;
}
curLevel--;// 当前层的节点数减一
// 当前层的所有节点处理完了
if (curLevel == 0) {
curLevel = nextLevel;
nextLevel = 0;
ans.add(tmp);
tmp = new ArrayList<>();
}
}
return ans;
}
}
107、二叉树的层次遍历II
给定一个二叉树,返回其节点值自底向上的层次遍历。 (即按从叶子节点所在层到根节点所在的层,逐层从左向右遍历)
例如:
给定二叉树 [3,9,20,null,null,15,7],
3
/ \
9 20
/ \
15 7
返回其自底向上的层次遍历为:
[
[15,7],
[9,20],
[3]
]
链接:https://leetcode-cn.com/problems/binary-tree-level-order-traversal-ii
题解
1、LinkedList 头插法,加入链表
/**
* 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) {
LinkedList<List<Integer>> ans = new LinkedList<List<Integer>>();
Queue<TreeNode> queue = new LinkedList<>();
if (root == null){
return ans;
}
queue.offer(root);
List<Integer> tmp = new ArrayList<>();
int nextLevel = 0;// 下一个层节点的数量
int curLevel = 1;// 当前节点的数量
while(!queue.isEmpty()) {
TreeNode node = queue.peek();
queue.poll();
tmp.add(node.val);
if (node.left != null) {
queue.offer(node.left);
nextLevel++;
}
if (node.right != null) {
queue.offer(node.right);
nextLevel++;
}
curLevel--;// 当前层的节点数减一
// 当前层的所有节点处理完了
if (curLevel == 0) {
curLevel = nextLevel;
nextLevel = 0;
ans.addFirst(tmp);
tmp = new ArrayList<>();
}
}
return ans;
}
}
转载请注明来源,欢迎对文章中的引用来源进行考证,欢迎指出任何有错误或不够清晰的表达。可以在下面评论区评论,也可以邮件至 1056615746@qq.com