LeetCode 103. 二叉树的锯齿形层次遍历 BFS

给定一个二叉树,返回其节点值的锯齿形层次遍历。(即先从左往右,再从右往左进行下一层遍历,以此类推,层与层之间交替进行)。

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

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

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


需要判断遍历的奇偶层,对于层序遍历的偶数层序列需要对结果进行一个反转( Collections.reverse(list);)再存储。


/**
 * Definition for a binary tree node.
 * public class TreeNode {
 *     int val;
 *     TreeNode left;
 *     TreeNode right;
 *     TreeNode(int x) { val = x; }
 * }
 */
class Solution {
    public List<List<Integer>> zigzagLevelOrder(TreeNode root) {
         
          List<List<Integer>> lists =  new ArrayList<>();

          if (root==null) return  lists;

        Queue<TreeNode> queue = new LinkedList<>();
        queue.add(root);
        int level = 1;
        while (!queue.isEmpty()){

            int count = queue.size();

            List<Integer> list = new ArrayList<>();

            while (count > 0){
                TreeNode temp = queue.poll();
                list.add(temp.val);
                    if (temp.left!=null){
                        queue.add(temp.left);
                    }

                    if (temp.right!=null){
                        queue.add(temp.right);
                    }



                count--;

            }

            if (level%2==0){
                Collections.reverse(list);
            }
            lists.add(list);
            level++;


        }

        return  lists;


    }
}


    原文作者:lhsjohn
    原文地址: https://www.jianshu.com/p/6ae8434cf6e8
    本文转自网络文章,转载此文章仅为分享知识,如有侵权,请联系博主进行删除。
点赞