剑指 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新特性,也可以直接写循环复制
代码:
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();
}
}
|