二叉樹的鋸齒形層次遍歷

給定一個二叉樹,返回其節點值的鋸齒形層次遍歷。(即先從左往右,再從右往左進行下一層遍歷,以此類推,層與層之間交替進行)。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

相關文章
相關標籤/搜索