题目:从上往下打印出二叉树的每个节点,同层节点从左至右打印。
——来源于《剑指offer》
- 思路:借助队列这种数据结构先进先出的特性,先将根结点的左右孩子压入队列,然后依次将子树的左右孩子压入队列。
- 代码实现:
import java.util.*;
/**
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<>();
if(root == null){
return list;
}
Queue<TreeNode> queue = new LinkedList<>();
TreeNode b = root;
queue.add(b);
while (!queue.isEmpty()) {
TreeNode treeNode = queue.poll();
list.add(treeNode.val);
if (treeNode.left != null) {
queue.add(treeNode.left);
}
if (treeNode.right != null) {
queue.add(treeNode.right);
}
}
return list;
}
}