【在线编程】从上往下打印二叉树
【问题描述】
从上往下打印出二叉树的每个节点,同层节点从左至右打印。
【解题思路 & Java实现】
方法:树的层次遍历。借助辅助队列,每次存储树的一层,然后读取队列的第一个节点,将值存到结果ArrayList中去,并继续将该节点的下一层加入到队列中。
import java.util.ArrayList;
/**
public class TreeNode {
int val = 0;
TreeNode left = null;
TreeNode right = null;
public TreeNode(int val) {
this.val = val;
}
}
*/
import java.util.*;
public class Solution {
public ArrayList<Integer> PrintFromTopToBottom(TreeNode root) {
ArrayList<Integer> result = new ArrayList<>();
if (root == null) {
return result;
}
Queue<TreeNode> nodes = new LinkedList<>();
nodes.offer(root);
while (!nodes.isEmpty()) {
TreeNode target = nodes.poll();
result.add(target.val);
//向队列中添加 target 的左孩子
if (target.left != null) {
nodes.offer(target.left);
}
//向队列中添加 target 的右孩子
if (target.right != null) {
nodes.offer(target.right);
}
}
return result;
}
}