103. 二叉树的锯齿形层序遍历

小豆丁 1年前 ⋅ 898 阅读
103. 二叉树的锯齿形层序遍历
给定一个二叉树,返回其节点值的锯齿形层序遍历。(即先从左往右,再从右往左进行下一层遍历,以此类推,层与层之间交替进行)。

例如:
给定二叉树 [3,9,20,null,null,15,7],

    3
   / \
  9  20
    /  \
   15   7
返回锯齿形层序遍历如下:

[
  [3],
  [20,9],
  [15,7]
]

解析

这个题目和普通的层序遍历一样,只不过我们需要控制一个变量,每次变换插入方向即可。

class Solution {
    public List<List<Integer>> zigzagLevelOrder(TreeNode root) {
        //层序遍历
        if(root==null) return new ArrayList<>();
        List<List<Integer>> res = new ArrayList<>();
        TreeNode head = root;
        boolean flag = true;
        LinkedList<TreeNode> queue = new LinkedList<>();
        queue.add(head);
        while(!queue.isEmpty()){
            int size = queue.size();
            LinkedList<Integer> lst = new LinkedList<>();
            for(int i = 0 ;i < size;i++){
                head = queue.removeFirst();
                if(flag){
                    lst.add(head.val);
                }else{
                    lst.addFirst(head.val);
                }
                if(head.left!=null)queue.add(head.left);
                if(head.right!=null)queue.add(head.right);
            }
            flag = !flag;
            res.add(lst);
        }

        return res;
    }
}