JZ59.按之字形顺序打印二叉树

 

请实现一个函数按照之字形打印二叉树，即第一行按照从左到右的顺序打印，第二层按照从右至左的顺序打印，第三行按照从左到右的顺序打印，其他行以此类推。

法1：

使用层次遍历（BFS），每层遍历时，当前出队节点入ans数组，判断当前层是奇数层还是偶数层，偶数层插入到ans数组最前面。

ps：判断层，每层遍历之前获得一下当前队列中节点个数size，每次遍历size--，size为0则遍历完一层。

遍历完一层将ans加入到res数组。

import java.util.ArrayList;
import java.util.*;
/*
public class TreeNode {
    int val = 0;
    TreeNode left = null;
    TreeNode right = null;

    public TreeNode(int val) {
        this.val = val;

    }

}
*/
public class Solution {
    public ArrayList<ArrayList<Integer> > Print(TreeNode pRoot) {
        ArrayList<ArrayList<Integer>> res = new ArrayList<>();
        if(pRoot==null) return res;
        Queue<TreeNode> q = new LinkedList<TreeNode>();
        q.offer(pRoot);
        int level = 1;
        while(!q.isEmpty()){
            int sz = q.size();
            ArrayList<Integer> ans = new ArrayList<>();
            while(sz-- > 0){
                TreeNode node = q.poll();
                if(level%2==0)
                    ans.add(0,node.val);
                else
                    ans.add(node.val);
             
                if(node.left!=null) q.offer(node.left);
                if(node.right!=null) q.offer(node.right);
            }
            level++;
            res.add(ans);
        }
        return res;
    }

}

注：BFS模板

不需要确定当前遍历到哪一层：

void bfs() {
 vis[] = {0}; // or set
 queue<int> pq(start_val);

 while (!pq.empty()) {
     int cur = pq.front(); pq.pop();
     for (遍历cur所有的相邻节点nex) {
         if (nex节点有效 && vis[nex]==0){
             vis[nex] = 1;
             pq.push(nex)
         }
     } // end for
 } // end while
}

需要确定当前遍历到哪一层：

void bfs() {
 int level = 0;
 vis[] = {0}; // or set
 queue<int> pq(original_val);
 while (!pq.empty()) {
     int sz = pq.size();

     while (sz--) {
             int cur = pq.front(); pq.pop();
         for (遍历cur所有的相邻节点nex) {
             if (nex节点有效 && vis[nex] == 0) {
                 vis[nex] = 1;
                 pq.push(nex)
             }
         } // end for
     } // end inner while
     level++;

 } // end outer while
}