剑指 Offer 32 - I. 从上到下打印二叉树
从上到下打印出二叉树的每个节点,同一层的节点按照从左到右的顺序打印。
例如: 给定二叉树: [3,9,20,null,null,15,7],
[3,9,20,15,7]链接:https://leetcode-cn.com/problems/cong-shang-dao-xia-da-yin-er-cha-shu-lcof
思路:
经典的层序遍历
因为LinkedList有堆栈的特性,
根节点root进入队列,然后弹出,取出value储存
将root的左节点和右节点先后进入队列
将左节点弹出,取出value储存,将其的左节点和右节点先后进入队列,
将右节点弹出,取出value储存,将其的左节点和右节点先后进入队列,
进行循环,达到层序遍历。
最后一行代码:将Integer类型list复制为int数组,使用了java8的stram新特性,也可以直接写循环复制
代码:
/**
* 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];
LinkedList<TreeNode> list = new LinkedList();
ArrayList<Integer> ints = new ArrayList();
list.push(root);
while (!list.isEmpty()){
TreeNode treeNode = list.removeLast();
ints.add(treeNode.val);
if(treeNode.left!=null)list.push(treeNode.left);
if(treeNode.right!=null)list.push(treeNode.right);
}
return ints.stream().mapToInt(k->k).toArray();
}
}