题目描述
题目链接:按之字形顺序打印二叉树_牛客题霸_牛客网 (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
输入:
{1,2,3,#,#,4,5}返回值:
[[1],[3,2],[4,5]]说明:
如题面解释,第一层是根节点,从左到右打印结果,第二层从右到左,第三层从左到右。
示例2
输入:
{8,6,10,5,7,9,11}返回值:
[[8],[10,6],[5,7,9,11]]
示例3
输入:
{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<ArrayList<Integer> > Print(TreeNode pRoot) {
ArrayList<ArrayList<Integer>> result = new ArrayList<>();
Queue<TreeNode> queue = new LinkedList<>();
queue.add(pRoot);
// 遍历标志位
boolean flag = true;
while(!queue.isEmpty()){
// 每次遍历前,先获取上一层有几个节点
int size = queue.size();
ArrayList<Integer> 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;
}
}