二叉树的层次遍历:
给你一个二叉树,请你返回其按 层序遍历 得到的节点值。 (即逐层地,从左到右访问所有节点)。
示例:
二叉树:[3,9,20,null,null,15,7],
3
/ \
9 20
/ \
15 7
返回其层序遍历结果:
[
[3],
[9,20],
[15,7]
]
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/binary-tree-level-order-traversal
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
层次遍历不难,我卡在了输出格式,菜鸡菜鸡。。
每次队列的长度其实就是这一层有几个节点。
vector<vector<int>> levelOrder(TreeNode* root) {
vector<vector<int>> res;
if(!root) return res;
queue<TreeNode*> q;
q.push(root);
while(!q.empty()) {
int size = q.size();
vector<int> v;
while(size--) {
TreeNode* t = q.front();
q.pop();
v.push_back(t->val);
if(t->left) q.push(t->left);
if(t->right) q.push(t->right);
}
res.push_back(v);
}
return res;
}
给定一个二叉树,返回其节点值的锯齿形层序遍历。(即先从左往右,再从右往左进行下一层遍历,以此类推,层与层之间交替进行)。
例如:
给定二叉树 [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
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
想明白就好了,如果是偶数层,我们需要从后向前看,在添加其子节点过程中,只需先看其右再看其左,头查即可;奇数层普通遍历;
public List<List<Integer>> zigzagLevelOrder(TreeNode root) {
Deque<TreeNode> deque = new LinkedList<>();
List<List<Integer>> res = new ArrayList<List<Integer>>();
if(root == null) return res;
deque.offerLast(root);
boolean flag = true;
while(!deque.isEmpty()) {
int size = deque.size();
List<Integer> list = new ArrayList<>();
while(size > 0) {
size--;
if(flag) {
TreeNode t = deque.pollFirst();
list.add(t.val);
if(t.left != null) deque.offerLast(t.left);
if(t.right != null) deque.offerLast(t.right);
}
else {
TreeNode t = deque.pollLast();
list.add(t.val);
if(t.right != null) deque.offerFirst(t.right);
if(t.left != null) deque.offerFirst(t.left);
}
}
flag = !flag;
res.add(new ArrayList<>(list));
}
return res;
}