牛客题霸NC14二叉树的之字形层序遍历Java题解
方法:利用队列
解题思路:将节点加入到队列中,利用队列Queue的先进后出,依此弹出节点。将每次弹出节点的值保存到一个ArrayList中(tmp),如果当前层是奇数层(从1开始),执行尾插,如果当前层是偶数层,执行头插。如果弹出的节点有左、右子节点,则将左、右子节点加入到队列Queue中,每一层遍历完,将这一层的节点都加入到ArrayList<ArrayList<integer>> list 中。
import java.util.*;
/*
* public class TreeNode {
* int val = 0;
* TreeNode left = null;
* TreeNode right = null;
* }
*/
public class Solution {
/**
*
* @param root TreeNode类
* @return int整型ArrayList<ArrayList<>>
*/
public ArrayList<ArrayList<Integer>> zigzagLevelOrder (TreeNode root) {
ArrayList<ArrayList<Integer>> res = new ArrayList<>();
Queue<TreeNode> queue = new LinkedList<>();
if(root!=null){
queue.add(root);
}
while(!queue.isEmpty()){
ArrayList<Integer> tmp = new ArrayList<>(); //存储每一层节点
for(int i= queue.size();i>0;i--){ //遍历当前层的节点
TreeNode node = queue.poll(); //弹出队列中的节点
if((res.size()+1)%2!=0){ //res.size()+1:当前的层数,从1开始
tmp.add(node.val); // 奇数层 -> 尾插
}else{
tmp.add(0,node.val); // 偶数层 -> 头插
}
if(node.left!=null){ //如果左子节点不为空,则将其加入到队列中
queue.add(node.left);
}
if(node.right!=null){ //如果左子节点不为空,则将其加入到队列中
queue.add(node.right);
}
}
res.add(tmp); //将这一层的节点加入到res中
}
return res;
}
}