【剑指offer-22】从上往下打印二叉树
- 考点:队列 树
- 时间限制:1秒
- 空间限制:32768K
- 从上往下打印出二叉树的每个节点,同层节点从左至右打印。
思路:
我看到题目大概知道这个考的就是层次遍历的做法了。
层次遍历通过队列进行一层层推进队列,然后一层层取出来。
在这里要注意一开始root为空的状况。
这道题比较简单,有思路,自己也做出来了。但是不太熟悉java的Queue接口。
代码:
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) {
TreeNode p = root;
Queue<TreeNode> queue = new LinkedList<TreeNode>();
queue.offer(p);
ArrayList<Integer> list = new ArrayList<Integer>();
if (p == null) {
return list;
}
while (queue.size() != 0) {
TreeNode q = queue.remove();
if (q.left != null) {
queue.offer(q.left);
}
if (q.right != null) {
queue.offer(q.right);
}
list.add(q.val);
}
return list;
}
}
我的问题:
- 首先是queue实例化的问题,在参考了下面的Java中Queue的用法后,得知不可以直接实例化Queue,反正牛客网上会报错,说什么抽象类之类的。
所以要使用:Queue<String> queue = new LinkedList<String>()
- 其次就是Queue的使用方法问题了:
- 添加元素用offer方法。【另一种是add,但是会有错误时候会抛出异常】
- 移除元素使用poll方法。【另一种是remove方法,同样有错误时候会抛出异常】
- peek方法用于查询队首元素。【另一种是element方法,同样也会抛出异常】
- 我在写算法的时候,犯了一个很傻的错,导致我一直报错。就是我制定了Queue以及ArrayList保存的数据类型,但是我在添加元素的时候忘记了这个问题。导致一直塞不进去,并且在add方法处报错。
- 最后又是差错控制了,如果root为null,直接返回空的ArrayList。
其他思路1:
看了大家的参考答案,原来这道题考察的是广度优先算法呢。(BFS)
广搜的套路就是用一个队列保存将要搜索的这一层的元素,然后逐个搜索;
1、将第一个元素加入队列
2、队列不为空时取队首元素
3、将下一层元素加入队尾
4、调到第二步,直到队列为空
其实可以用ArrayList模拟Queue,这样就不至于考试的时候,还在查Queue的用法了。
/**
思路是用arraylist模拟一个队列来存储相应的TreeNode
*/
public class Solution {
public ArrayList<Integer> PrintFromTopToBottom(TreeNode root) {
ArrayList<Integer> list = new ArrayList<>();
ArrayList<TreeNode> queue = new ArrayList<>();
if (root == null) {
return list;
}
queue.add(root);
while (queue.size() != 0) {
// arrayList这里remove(0)就可以达到同样的移除队首的效果
TreeNode temp = queue.remove(0);
if (temp.left != null){
queue.add(temp.left);
}
if (temp.right != null) {
queue.add(temp.right);
}
list.add(temp.val);
}
return list;
}
}
- arrayList这里remove(0)就可以达到同样的移除队首的效果。