题目描述
从上往下打印出二叉树的每个节点,同层节点从左至右打印。
根据题目描述,其实就是求二叉树的层序遍历结果。
思路:遍历当前节点,使用队列来储存子节点,然后从队列取下一个要遍历的节点,将该节点的子节点放入队列...递归下去,最终得到遍历结果。
实现代码:
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> values = new ArrayList<Integer>(); //储存遍历结果
ArrayList<TreeNode> queue = new ArrayList<TreeNode>();//储存子节点
//判断是否为空树
if(root == null){
return values;
}
TreeNode node = root ; //当前节点
queue.add(node);
//队列中没有值时遍历结束
while(queue.size() != 0){
//从队列取出当前节点
node = queue.remove(0);
//当前节点的遍历结果
values.add(node.val);
//先把左子节点放进队列
if(node.left != null){
queue.add(node.left);
}
//然后把右子节点放进队列
if(node.right != null){
queue.add(node.right);
}
}
return values;
}
}