《剑指offer》—— 22. 从上往下打印二叉树(Java)
推荐
完整《剑指Offer》算法题解析系列请点击 👉 《剑指Offer》全解析 Java 版
题目描述
从上往下打印出二叉树的每个节点,同层节点从左至右打印。
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) {
}
}
思路:
就是使用队列来进行层次遍历;
首先了解一下队列:
队列是一种特殊的线性表,它只允许在表的前端进行删除操作,而在表的后端进行插入操作。
LinkedList类实现了Queue接口,因此我们可以把LinkedList当成Queue来用。
队列的几个常用方法:
- poll() ,返回队列中的第一个元素,并且在队列中删除这个元素。
- peek(),返回队列中的第一元素,队列为空时,返回null。
- element() ,返回队列中的第一元素,队列为空时,抛出一个异常。
- add(),向队列中添加一个元素,失败时会抛出异常。(不推荐)
- offer(),向队列中添加一个元素,失败时会返回false。(推荐)
具体思路可看代码中注释。
实现:
import java.util.ArrayList;
import java.util.Queue;
import java.util.LinkedList;
/**
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<Integer>();
if (root == null){
return list;
}
Queue<TreeNode> queue = new LinkedList<TreeNode>();
// 1.将树的的根结点放入队列
queue.offer(root);
// 2.队列不为空时执行循环3、4、5、6
while(!queue.isEmpty()){
// 3.将队列中的第一个元素作为树的当前结点,并在队列中删除这个元素;
TreeNode t = queue.poll();
// 4. 将树的当前节点的 左节点放入队列
if (t.left != null){
queue.offer(t.left);
}
// 5. 将树的当前节点的 右节点放入队列
if (t.right != null){
queue.offer(t.right);
}
// 6.将当前节点存入list集合中
list.add(t.val);
}
return list;
}
}
看完之后,如果还有什么不懂的,可以在评论区留言,会及时回答更新。
这里是猿兄,为你分享程序员的世界。
非常感谢各位大佬们能看到这里,如果觉得文章还不错的话, 求点赞👍 求关注💗 求分享👬求评论📝 这些对猿兄来说真的 非常有用!!!
注: 如果猿兄这篇博客有任何错误和建议,欢迎大家留言,不胜感激!