题目链接:按之字形顺序打印二叉树_牛客题霸_牛客网 (nowcoder.com)
给定一个二叉树,返回该二叉树的之字形层序遍历,(第一层从左向右,下一层从右向左,一直这样交替)
数据范围:0 \le n \le 15000≤n≤1500,树上每个节点的val满足 |val| <= 1500∣val∣<=1500
要求:空间复杂度:O(n)O(n),时间复杂度:O(n)O(n)例如:
给定的二叉树是{1,2,3,#,#,4,5}该二叉树之字形层序遍历的结果是
[
[1],
[3,2],
[4,5]
]
输入:
{1,2,3,#,#,4,5}返回值:
[[1],[3,2],[4,5]]说明:
如题面解释,第一层是根节点,从左到右打印结果,第二层从右到左,第三层从左到右。
输入:
{8,6,10,5,7,9,11}返回值:
[[8],[10,6],[5,7,9,11]]
输入:
{1,2,3,4,5}返回值:
[[1],[3,2],[4,5]]
刚读完题目,容易被题目吓着,之字形是怎么弄,我只会前序 、中序、后序遍历,没听过之字形遍历。看完示例,意思就是 先从左到右,再从右到左,那怎么弄呢,第一行我直接加入然后遍历,第二行 我先加右子树,再加左子树,这样行吗? 如果这层只有一颗子树,这样是可以的,如果有两颗及以上子树,这样做,就只是局部从右向左遍历。所以这个解题思路也不行,那怎么做呢,且听下回分解!
开玩笑开玩笑! 那就说下我们的解题思路,首先用队列加入根节点,然后在循环里,把它取出来,取出来以后,把它的值加入临时链表,然后把它的左右子树加入队列中,但是我们怎么知道,它在这层有几个子树呢,我们不是有个队列嘛,每次在从它里面取出上一层加入到里面的节点前,队列有个大小,这个大小就是上一层的节点数,所以用这个节点数控制,也就是再加一个循环,进行每一层的节点遍历。 思路一说倒是很容易。
但是易错,易错点有下面几个,一个是 我用布尔变量控制方向,怎么样不停的换true和false呢,通过 flag = !flag 这种方式来设置标志位。 第二个是,我如何让它出现从左到右,再从右到左呢, 这里就是用 add() 这个方法的参数,如果直接放值,就是从左到右的遍历顺序,但如果是用两个参数的 add(index,value),第一个参数是要放到下标,我们用 add(0,value),让后遍历的放在链表的前面,就能出现从右往左遍历的效果。 第三个是,
如果读到二叉树的最后一层,那它们的子树肯定都是空,临时链表 里面就没有数据,这时候如果加入二维链表,就会产生一个空链表,所以要判断 临时链表里的数据不为0,才加入二维链表。
- import java.util.*;
- import java.util.ArrayList;
-
- /*
- public class TreeNode {
- int val = 0;
- TreeNode left = null;
- TreeNode right = null;
- public TreeNode(int val) {
- this.val = val;
- }
- }
- */
- public class Solution {
- public ArrayList
> Print(TreeNode pRoot) { -
- ArrayList
> result = new ArrayList<>(); - Queue
queue = new LinkedList<>(); - queue.add(pRoot);
- // 遍历标志位
- boolean flag = true;
- while(!queue.isEmpty()){
- // 每次遍历前,先获取上一层有几个节点
- int size = queue.size();
-
- ArrayList
list = new ArrayList<>(); -
- for(int i=0;i< size;i++){
- TreeNode node = queue.poll();
- // 如果读到的节点为空,就跳过本次循环 ,对根节点为空也有效
- if(node == null)
- continue;
- if(flag)
- list.add(node.val);
- else{
- list.add(0,node.val);
- }
- queue.add(node.left);
- queue.add(node.right);
-
- }
- if(list.size()!=0)
- result.add(list);
- flag = !flag;
- }
- return result;
- }
- }