层次遍历应当很熟的啦~~~
import java.util.*;
public class Solution {
public ArrayList<ArrayList<Integer>> levelOrder (TreeNode root) {
// write code here
// 练下层次遍历
ArrayList<ArrayList<Integer>> res = new ArrayList<ArrayList<Integer>>();
ArrayDeque<TreeNode> dq = new ArrayDeque<>();
if(root==null)
return res;
dq.add(root);
while(!dq.isEmpty()){
int curSize = dq.size();
ArrayList<Integer> arr = new ArrayList<>();
while(curSize-->0){
TreeNode node = dq.pollFirst();
arr.add(node.val);
if(node.left!=null) dq.add(node.left);
if(node.right!=null) dq.add(node.right);
}
res.add(arr);
}
return res;
}
}