版权声明:博客内容为本人自己所写,请勿转载。 https://blog.csdn.net/weixin_42805929/article/details/83107180
从上往下打印二叉树 java
题目描述
从上往下打印出二叉树的每个节点,同层节点从左至右打印。
思路:
按照层次遍历的方法,使用队列辅助。
1.将根结点加入队列。
2.循环出队,打印当前元素,若该结点有左子树,则将其加入队列,若有右子树,将其加入队列。
3.直到队列为空,表明已经打印完所有结点。
代码1:
import java.util.*;
public class Solution {
public ArrayList<Integer> PrintFromTopToBottom(TreeNode root) {
ArrayList<Integer> al = new ArrayList<>();
ArrayList<TreeNode> queue = new ArrayList<>();
if(root == null){
return al;
}
queue.add(root);
while(queue.size() != 0){
TreeNode temp = queue.remove(0);
al.add(temp.val);
if(temp.left != null){
queue.add(temp.left);
}
if(temp.right != null){
queue.add(temp.right);
}
}
return al;
}
}
代码2:
public class Solution {
public ArrayList<Integer> PrintFromTopToBottom(TreeNode root) {
ArrayList<Integer> al = new ArrayList<>();
Queue<TreeNode> queue = new LinkedList<>();
if(root == null){
return al;
}
queue.offer(root);
while(!queue.isEmpty()){
TreeNode temp = queue.poll();
if(temp.left != null){
queue.offer(temp.left);
}
if(temp.right != null){
queue.offer(temp.right);
}
al.add(temp.val);
}
return al;
}
}