从上往下打印出二叉树的每个节点,同层节点从左至右打印
二叉树的层次遍历,实际也是广度优先遍历。一般的解题套路是使用队列来保存将要搜索的这一层元素,然后逐个查看。元素入队的规则满足先进先出原则
使用 ArrayList 或者 LinkedList 都可以实现上述功能
/**
* 使用 ArrayList 实现队列
*/
public class Solution {
public ArrayList<Integer> PrintFromTopToBottom(TreeNode root) {
ArrayList<Integer> result = new ArrayList<>();
ArrayList<TreeNode> queue = new ArrayList<>();
if(root == null) {
return result;
}
queue.add(root);
while(!queue.isEmpty()) {
TreeNode node = queue.remove(0);
if(node.left != null) {
queue.add(node.left);
}
if(node.right != null) {
queue.add(node.right);
}
result.add(node.val);
}
return result;
}
}
/**
* 使用 ArrayList 实现队列
*/
import java.util.LinkedList;
public class Solution {
public ArrayList<Integer> PrintFromTopToBottom(TreeNode root) {
ArrayList<Integer> result = new ArrayList<>();
LinkedList<TreeNode> queue = new LinkedList<>();
if(root == null) {
return result;
}
queue.offer(root);
while(!queue.isEmpty()) {
TreeNode node = queue.poll();
if(node.left != null) {
queue.offer(node.left);
}
if(node.right != null) {
queue.offer(node.right);
}
result.add(node.val);
}
return result;
}
}
相关文章
暂无评论...