按之字形顺序打印二叉树
请实现一个函数按照之字形打印二叉树,即第一行按照从左到右的顺序打印,第二层按照从右至左的顺序打印,第三行按照从左到右的顺序打印,其他行以此类推。
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>> layers = new ArrayList<ArrayList<Integer>>(); if (pRoot == null) return layers; Deque<TreeNode> queue = new LinkedList<TreeNode>(); queue.offer(pRoot); int depth = 0; while (!queue.isEmpty()) { depth ++; ArrayList<Integer> layer = new ArrayList<Integer>(); int cur = 0, size = queue.size(); if ((depth & 1) == 0) { //如果是偶数层倒序添加 Iterator<TreeNode> it = queue.descendingIterator(); while (it.hasNext()) { layer.add(it.next().val); } } else { //如果是奇数层正序添加 Iterator<TreeNode> it = queue.iterator(); while (it.hasNext()) { layer.add(it.next().val); } } while (cur < size) { TreeNode node = queue.poll(); if (node.left != null) { queue.offer(node.left); } if (node.right != null) { queue.offer(node.right); } cur ++; } layers.add(layer); } return layers; } }
作者:Pickle
声明:对于转载分享我是没有意见的,出于对博客园社区和作者的尊重一定要保留原文地址哈。
致读者:坚持写博客不容易,写高质量博客更难,我也在不断的学习和进步,希望和所有同路人一道用技术来改变生活。觉得有点用就点个赞哈。