【树】剑指 Offer 32 - III. 从上到下打印二叉树 III
题目:
请实现一个函数按照之字形顺序打印二叉树,即第一行按照从左到右的顺序打印,第二层按照从右到左的顺序打印,第三行再按照从左到右的顺序打印,其他行以此类推。
例如:
给定二叉树: [3,9,20,null,null,15,7],
3
/ \
9 20
/ \
15 7
返回其层次遍历结果:
[
[3],
[20,9],
[15,7]
]
解答:
方法一:
在剑指 Offer 32 - II. 从上到下打印二叉树 II解答的基础上增加一个isOdd变量,当是奇数层时从左到右遍历,否则从右到左遍历。
/** * Definition for a binary tree node. * public class TreeNode { * int val; * TreeNode left; * TreeNode right; * TreeNode(int x) { val = x; } * } */ class Solution { public List<List<Integer>> levelOrder(TreeNode root) { ArrayList<List<Integer>> res = new ArrayList<>(); if(root == null) return res; Queue<TreeNode> queue = new LinkedList<>(); queue.add(root); boolean isOdd = true; while(!queue.isEmpty()){ ArrayList<Integer> curLevel = new ArrayList<>(); for(int i = queue.size();i>0;i--){ TreeNode node = queue.poll(); curLevel.add(node.val); if(node.left!=null) queue.add(node.left); if(node.right!=null) queue.add(node.right); } if(isOdd) res.add(curLevel);
//偶数层倒叙,还可以用Collections.reverse(tmp); if(!isOdd){ ArrayList<Integer> tmp = new ArrayList<>(); for(int i = curLevel.size()-1;i>=0;i--){ tmp.add(curLevel.get(i)); } res.add(tmp); } isOdd = !isOdd; } return res; } }
用 Collections.reverse 反转一个 List
https://www.cnblogs.com/ywb2018/p/9922829.html
方法二:层序遍历+双端队列
class Solution { public List<List<Integer>> levelOrder(TreeNode root) { Queue<TreeNode> queue = new LinkedList<>(); List<List<Integer>> res = new ArrayList<>(); if(root != null) queue.add(root); while(!queue.isEmpty()) { LinkedList<Integer> tmp = new LinkedList<>(); for(int i = queue.size(); i > 0; i--) { TreeNode node = queue.poll(); if(res.size() % 2 == 0) tmp.addLast(node.val); // 偶数层 -> 队列头部 else tmp.addFirst(node.val); // 奇数层 -> 队列尾部 if(node.left != null) queue.add(node.left); if(node.right != null) queue.add(node.right); } res.add(tmp); } return res; } }

浙公网安备 33010602011771号