题解 | #按之字形顺序打印二叉树# 层序遍历结果反转
按之字形顺序打印二叉树
https://www.nowcoder.com/practice/91b69814117f4e8097390d107d2efbe0
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 { /** * 代码中的类名、方法名、参数名已经指定,请勿修改,直接返回方法规定的值即可 * * * @param pRoot TreeNode类 * @return int整型ArrayList<ArrayList<>> */ public ArrayList<ArrayList<Integer>> Print (TreeNode root) { if (root == null) return new ArrayList<>(); // write code here Queue<TreeNode> q1 = new ArrayDeque<>(); Queue<TreeNode> q2 = new ArrayDeque<>(); q1.add(root); ArrayList<ArrayList<Integer>> result = new ArrayList<>(); while (!q1.isEmpty() || !q2.isEmpty()) { ArrayList<Integer> arr = new ArrayList<>(); if (!q1.isEmpty()) { while (!q1.isEmpty()) { TreeNode t = q1.poll(); arr.add(t.val); if (t.left != null) q2.add(t.left); if (t.right != null) q2.add(t.right); } } else if (!q2.isEmpty()) { while (!q2.isEmpty()) { TreeNode t = q2.poll(); arr.add(t.val); if (t.left != null) q1.add(t.left); if (t.right != null) q1.add(t.right); } } result.add(arr); } // 直接操作结果 int len = result.size(); for (int i = 0; i < len; i++) { ArrayList<Integer> t = result.get(i); if (i % 2 == 1) { Collections.reverse(t); } } return result; } }
按照正常层序遍历处理,查看 NC15 求二叉树的层序遍历 代码,然后将其结果,按照层数奇偶,决定是否反转即可。