@author: sdubrz
@date: 8/2/2020 9:24:21 AM
难度: 中等
考察内容: 二叉树
@e-mail: lwyz521604#163.com
题目来自《剑指offer》 电子工业出版社
从上到下打印出二叉树的每个节点,同一层的节点按照从左到右的顺序打印。
例如:
给定二叉树: [3,9,20,null,null,15,7],
3
/ \
9 20
/ \
15 7
返回:
[3,9,20,15,7]
提示:
节点总数 <= 1000
解法
这个问题其实是一个广度优先搜索,可以用一个队列来实现:
/**
* Definition for a binary tree node.
* public class TreeNode {
* int val;
* TreeNode left;
* TreeNode right;
* TreeNode(int x) { val = x; }
* }
*/
class Solution {
public int[] levelOrder(TreeNode root) {
if(root==null) {
return new int[0];
}
Queue<TreeNode> queue = new LinkedList<>();
ArrayList<Integer> list = new ArrayList<>();
queue.add(root);
while(!queue.isEmpty()) {
TreeNode current = queue.poll();
list.add(current.val);
if(current.left!=null) {
queue.add(current.left);
}
if(current.right!=null) {
queue.add(current.right);
}
}
int[] result = new int[list.size()];
Iterator<Integer> iter = list.iterator();
int index = 0;
while(iter.hasNext()) {
result[index] = (int) iter.next();
index++;
}
return result;
}
}
在 LeetCode 系统中提交的结果为
执行结果:通过 显示详情
执行用时:1 ms, 在所有 Java 提交中击败了99.65%的用户
内存消耗:40 MB, 在所有 Java 提交中击败了35.23%的用户