题目
给定一个二叉树,返回其节点值的锯齿形层序遍历。(即先从左往右,再从右往左进行下一层遍历,以此类推,层与层之间交替进行)。
思路
其实就是层次遍历,加个数值去判断应该在列表进行头插法还是尾插法就可以了,简答题!
代码
class Solution {
public List<List<Integer>> zigzagLevelOrder(TreeNode root) {
List<List<Integer>> ans = new LinkedList<List<Integer>>();
if(root==null)
return ans;
Deque<TreeNode> deque = new LinkedList<>();
deque.add(root);
int num=0;// 偶数从左到右遍历,奇数从右到左遍历
while (!deque.isEmpty())
{
int size=deque.size();
List<Integer> list =new LinkedList<>();
for (int i=size;i>0;--i)
{
TreeNode poll = deque.poll();
if (poll.left!=null)
deque.add(poll.left);
if (poll.right!=null)
deque.add(poll.right);
if ((num&1)==0) // 偶数,从右往左遍历
{
list.add(poll.val);
}
else
{
list.add(0, poll.val);
}
}
ans.add(list);
++num;//改变遍历顺序
}
return ans;
}
}
结果
本来想利用一下双端队列的特性的,但是仔细一想,每个节点可能会有子节点,加入进来后,会使得双端队列的一端失效(也就是有了子节点后不能弹出,因为那部分是下一层的),最后利用了链表的头插法来解决问题!