二叉树的锯齿形层次遍历

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

例如:bash

给定二叉树 [3,9,20,null,null,15,7],ui

3
   / \
  9  20
    /  \
   15   7
复制代码

返回锯齿形层次遍历以下:spa

[
  [3],
  [20,9],
  [15,7]
]
复制代码
/**
 * 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>> res = new ArrayList<List<Integer>>();
        if( root == null ) return res;
        Deque<TreeNode> tree = new LinkedList<>();
        tree.add( root );
        TreeNode end = root;
        Boolean flag = true;
        List<Integer> list = new ArrayList<>();
        while( !tree.isEmpty() ){
          TreeNode cur = flag ?  tree.pollFirst() : tree.pollLast();
          list.add( cur.val );
          if( flag ){
              if( cur.left != null ){
                 tree.offerLast( cur.left );
              }
              if( cur.right != null ){
                  tree.offerLast( cur.right );
              }
          
          }else{
              if( cur.right != null ){
                  tree.offerFirst( cur.right );
              }
              if( cur.left != null ){
                 tree.offerFirst( cur.left );
              } 
          }  
          if( cur == end ){
              res.add( list );
              list = new ArrayList<>();
              end = flag ? tree.peekFirst() : tree.peekLast();
              flag = !flag;
          }  
        }
        return  res;
    }
}
复制代码

解题思路: 使用双向链表保存下一层的值 , 用end指向当前层遍历的最后一个结点,若是当前层的遍历顺序是从左到右 , 则每次弹出的都是链表的第一个元素 , 而后再把左右结点压入链表的尾部,先压左再压右;若是当前层的遍历顺序是从右到左,则每次弹出的都是最后一个结点 , 而后把右左压入到头部,先压右再压左。code

相关文章
相关标签/搜索