用ArrayList来模拟队列,存放节点。
remove()删除并返回值
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<Integer> PrintFromTopToBottom(TreeNode root) {
ArrayList<Integer> list = new ArrayList<>();
ArrayList<TreeNode> queue = new ArrayList<>();
//判断边界值,增加代码鲁棒性
if(root == null){
return list;
}
//引入一个队列来储存节点值
queue.add(root);
while(queue.size() != 0){
//删除queue的第一个元素并将其返回,模拟了队列的弹出操作
TreeNode temp = queue.remove(0);
//储存左子树根节点
if(temp.left != null){
queue.add(temp.left);
}
//储存右子树根节点
if(temp.right != null){
queue.add(temp.right);
}
//打印的列表即为temp的值
list.add(temp.val);
}
return list;
}
}