思路
相较于二叉树的层序遍历,多了一个flag变量,当flag等于0时,把当前层的数组从左到右放入链表,当flag等于1时,把当前层的数组从右到左放入链表。
解题方法
注意:链表删除一个数据后会立即重排,所以删除同一层的节点时,每次都删除第一个节点。
Code
/**
* Definition for a binary tree node.
* public class TreeNode {
* int val;
* TreeNode left;
* TreeNode right;
* TreeNode() {}
* TreeNode(int val) { this.val = val; }
* TreeNode(int val, TreeNode left, TreeNode right) {
* this.val = val;
* this.left = left;
* this.right = right;
* }
* }
*/
class Solution {
public List<List<Integer>> zigzagLevelOrder(TreeNode root) {
List<List<Integer>> res=new ArrayList<List<Integer>>();
if(root==null) return res;
List<TreeNode> queue = new ArrayList<TreeNode>();
queue.add(root);
int flag=0;
while(queue.size()>0){
int size=queue.size();
List<Integer> t = new ArrayList<Integer>();
if(flag==0){
for(int i=0;i<size;i++){
t.add(queue.get(i).val);
}
flag=1;
}else{
for(int i=size-1;i>=0;i--){
t.add(queue.get(i).val);
flag=0;
}
}
for(int i=0;i<size;i++){
if(queue.get(i).left!=null){
queue.add(queue.get(i).left);
}
if(queue.get(i).right!=null){
queue.add(queue.get(i).right);
}
}
for(int i=0;i<size;i++){
queue.remove(0);
}
res.add(t);
}
return res;
}
}