[剑指OFFER]之树：JZ77 按之字形顺序打印二叉树

1.题目

给定一个二叉树，返回该二叉树的之字形层序遍历，（第一层从左向右，下一层从右向左，一直这样交替）

2.解法

2.1 树的结构

 class TreeNode {
        int val = 0;
        TreeNode left = null;
        TreeNode right = null;
        public TreeNode(int val) {
            this.val = val;
        }
    }

2.2 解法

使用Queue<TreeNode> queue = new ArrayDeque();来存储每一级的节点元素。
并且在遍历时，将这一级元素全部取出，并将下一级元素存入。
reverse是实现逆序控制的标志。

public ArrayList<ArrayList<Integer> > Print(TreeNode pRoot) {
        ArrayList<ArrayList<Integer>> ans = new ArrayList<>();
        Queue<TreeNode> queue = new ArrayDeque();
        if (pRoot == null) return ans;
        queue.offer(pRoot);
        boolean reverse = false;
        while (!queue.isEmpty()) {
            int size = queue.size();
            ArrayList<Integer> curlevel = new ArrayList<>();
            for (int i = 0; i < size; i++) {
                TreeNode node = queue.poll();
                if(reverse)
                    curlevel.add(0,node.val);
                else
                    curlevel.add(node.val);
                if (node.left != null) queue.offer(node.left);
                if (node.right != null) queue.offer(node.right);
            }
            reverse=!reverse;
            ans.add(curlevel);
        }
        return ans;
    }

总结

使用宽度优先遍历。

算法系列在github上有一个开源项目，主要是本系列博客的demo代码。https://github.com/forestnlp/alg
如果您对软件开发、机器学习、深度学习有兴趣请关注本博客，将持续推出Java、软件架构、深度学习相关专栏。
您的支持是对我最大的鼓励。