思路:
用队列的size来做循环,每次把队列中的元素弹出来后,加到list集合中,等循环结束后,list集合已经装好了一层的节点数值,再装入到res里面
核心代码:
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) {
// write code here
ArrayList<ArrayList<Integer>> res=new ArrayList<ArrayList<Integer>>();
Queue<TreeNode> queue=new LinkedList<>();
queue.add(root);
while(!queue.isEmpty()){
int size=queue.size();
ArrayList<Integer> list=new ArrayList<Integer>();
for(int i=0;i<size;i++){
TreeNode cur=queue.poll();
list.add(cur.val);
if(cur.left!=null){
queue.add(cur.left);
}
if(cur.right!=null){
queue.add(cur.right);
}
}
res.add(list);
}
return res;
}
}
如果别人要求返回的是二维数组,那需要将集合转为二维数组
Java中ArrayList转成二维数组以及int[]数组和ArrayList<Integer>转换