LeetCode 103: 二叉树的锯齿形层序遍历
题目
给你二叉树的根节点 root ,返回其节点值的 锯齿形层序遍历 。(即先从左往右,再从右往左进行下一层遍历,以此类推,层与层之间交替进行)。
示例 1:
输入:root = [3,9,20,null,null,15,7]
输出:[[3],[20,9],[15,7]]
示例 2:
输入:root = [1]
输出:[[1]]
示例 3:
输入:root = []
输出:[]
思路
树的层序遍历,使用双端队列,可以从队尾或队头存入元素
代码
class Solution {
public List<List<Integer>> zigzagLevelOrder(TreeNode root) {
List<List<Integer>> res=new ArrayList<>();
if(root==null)
return res;
Queue<TreeNode> que=new LinkedList<TreeNode>();
que.add(root);
while(!que.isEmpty())
{
List<Integer> l=new ArrayList<>();
int count=que.size();
while(count>0)
{
TreeNode t=que.poll();
if(res.size()%2==0)
l.add(t.val);
else
l.add(0,t.val);
if(t.left!=null)
que.add(t.left);
if(t.right!=null)
que.add(t.right);
count--;
}
res.add(l);
}
return res;
}
}
反思
①可以通过res的大小判断当前遍历的层数
②创建队列的语句:Queue