剑指offer - 按之字形顺序打印二叉树

题目链接

按之字形打印二叉树

算法思路

BFS算法框架
这题主要考查的是树的层次遍历，只要按层次遍历思路来遍历就行了！唯一不同是在添加到结果列表的时候，先判断一下是否要反转列表。通过奇偶数来判断是否要反转列表。

代码实现

import java.util.*;

/*
public class TreeNode {
    int val = 0;
    TreeNode left = null;
    TreeNode right = null;

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

    }

}
*/
import java.util.ArrayList;
import java.util.LinkedList;
import java.util.Queue;

public class Solution {

	//结果集列表
    ArrayList<ArrayList<Integer>> res = new ArrayList<>();
    public ArrayList<ArrayList<Integer> > Print(TreeNode pRoot) {
        if (pRoot == null) {
            return res;
        }
        levelTravers(pRoot);
        return res;
    }

	//层次遍历
    private void levelTravers(TreeNode pRoot) {
        Queue<TreeNode> q = new LinkedList<>();
        q.offer(pRoot);
        while (!q.isEmpty()) {
            int size = q.size();
            ArrayList<Integer> list = new ArrayList<>();
            for (int i = 0; i < size; i++) {
                TreeNode node = q.poll();
                list.add(node.val);
                if(node.left!=null){
                    q.offer(node.left);
                }
                if (node.right != null) {
                    q.offer(node.right);
                }
            }
            //如果是奇数，则反转list
            if((res.size() & 1) == 1 ){
                Collections.reverse(list);
            }
            res.add(list);
        }

    }

}