题目:从上往下打印出二叉树的每个节点,同层节点从左至右打印。
思路:从上往下打印,每层都是从左到右打印,显然就是按层打印,由于不需要进行换行,因此很简单,使用一个队列的数据结构就可以了。
常识:使用LinkedList等类库时注意引入相应的包。
//注意细节:使用LinkedList等类库时要注意引入相应的包
import java.util.*;
//即为按层遍历,不需要换行,使用队列来解决
public class Solution {
public ArrayList<Integer> PrintFromTopToBottom(TreeNode root) {
//要求返回的是list,因此先建立一个list用来存放答案
ArrayList<Integer> results=new ArrayList<>();
//特殊输入
if(root==null) return results;
//调用方法来遍历二叉树
this.process(root,results);
//返回结果
return results;
}
//按层遍历一棵二叉树
private void process(TreeNode root, ArrayList<Integer> list){
//①创建linkedList作为队列
LinkedList<TreeNode> queue=new LinkedList<>();
//②将根结点放入队列
queue.add(root);
//③循环:弹出--遍历次结点--先后压入左右结点
while(queue.size()!=0){
TreeNode temp=queue.poll();
list.add(temp.val);
if(temp.left!=null) queue.add(temp.left);
if(temp.right!=null) queue.add(temp.right);
}
}
}