题目
- 从上到下按层打印二叉树,同一层结点从左至右输出。每一层输出一行。
代码
import java.util.ArrayList;
import java.util.Queue;
import java.util.LinkedList;
/*
public class TreeNode {
int val = 0;
TreeNode left = null;
TreeNode right = null;
public TreeNode(int val) {
this.val = val;
}
}
*/
public class Solution {
ArrayList<ArrayList<Integer>> Print(TreeNode pRoot) {
ArrayList<ArrayList<Integer>> resultlist = new ArrayList<>();
ArrayList<Integer> templist = new ArrayList<>();
Queue<TreeNode> queue = new LinkedList<>();
if(pRoot==null)
return resultlist;
queue.add(pRoot);
while(queue.size()!=0){
int len = queue.size();
//每次templist集合中添加元素都需要清除元素,然后再添加
templist.clear();
for(int i=0;i<len;i++){
TreeNode temp =queue.poll();
templist.add(temp.val);
if(temp.left!=null){
queue.add(temp.left);
}
if(temp.right!=null){
queue.add(temp.right);
}
}
resultlist.add(new ArrayList<>(templist));
}
return resultlist;
}
}