剑指 Offer 32 - I. 从上到下打印二叉树(7.29)
剑指 Offer 32 - I. 从上到下打印二叉树
从上到下打印出二叉树的每个节点,同一层的节点按照从左到右的顺序打印。
例如:
给定二叉树: [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; }
* }
*/
//BFS,可以利用队列特性解题
class Solution {
public int[] levelOrder(TreeNode root) {
//根节点为空,直接返回空
if(root==null) return new int[0];
Queue<TreeNode> queue=new LinkedList<>();
//将根节点入队,然后开始循环
queue.add(root);
//保存值
List<Integer> list=new ArrayList();
while (!queue.isEmpty()){
//出队,保存值,左右节点非空则入队
TreeNode node=queue.poll();
list.add(node.val);
if(node.left!=null) queue.add(node.left);
if(node.right!=null) queue.add(node.right);
}
//将list转化为数组
int[] a=new int[list.size()];
for(int i=0;i<a.length;i++){
a[i]=list.get(i);
}
return a;
}
}
补充:
poll,peek,element的共同点:
都是返回队列中的首个元素
不同点:
poll:将首个元素从队列中弹出,如果队列是空的,就返回null
peek:查看首个元素,不会移除首个元素,如果队列是空的就返回null
element:查看首个元素,不会移除首个元素,如果队列是空的就抛出异常NoSuchElementException
题目来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/cong-shang-dao-xia-da-yin-er-cha-shu-lcof
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。