- 题目描述
- 思路以及解答
题目描述
从上往下打印出二叉树的每个节点,同层节点从左至右打印。
思路以及解答
主要的思路是借助队列,先把当前元素放进队列,然后不断取出元素,取出元素的时候,同时处理其左子树和右子树,将其放到队列中。直到队列是空的。
总所周知,队列的特点是先进先出,也就是我们可以保证,根节点先进去,然后根节点取出来,打印根节点,同时如果根节点的左节点不为空,则将左节点放进去队列,右节点如果不为空,同样放进去队列中,然后又取出队列的第一个元素,打印,将它的左右节点不为空的加到队列中,这样循环下去即可...
PS:LinkedList
是双向队列,queue.add()
是在队列后面添加元素,queue.pollFirst()
是将第一个元素取出队列。
import java.util.ArrayList;
import java.util.LinkedList;
import java.util.Queue;
/**
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> results = new ArrayList<>();
Queue<TreeNode> queue = new LinkedList<>();
if (root != null) {
queue.add(root);
while (!queue.isEmpty()) {
TreeNode treeNode = ((LinkedList<TreeNode>) queue).pollFirst();
results.add(treeNode.val);
if(treeNode.left!=null){
((LinkedList<TreeNode>) queue).add(treeNode.left);
}
if(treeNode.right!=null){
((LinkedList<TreeNode>) queue).add(treeNode.right);
}
}
}
return results;
}
}
【作者简介】: 秦怀,公众号【秦怀杂货店】作者,技术之路不在一时,山高水长,纵使缓慢,驰而不息。这个世界希望一切都很快,更快,但是我希望自己能走好每一步,写好每一篇文章,期待和你们一起交流。
此文章仅代表自己(本菜鸟)学习积累记录,或者学习笔记,如有侵权,请联系作者核实删除。人无完人,文章也一样,文笔稚嫩,在下不才,勿喷,如果有错误之处,还望指出,感激不尽~