剑指22 - 从上往下打印二叉树(层层从左到右)
题目描述
从上往下打印出二叉树的每个节点,同层节点从左至右打印。
示例1
输入
{5,4,#,3,#,2,#,1}
返回值
[5,4,3,2,1]
import java.util.ArrayList;
/**
public class TreeNode {
int val = 0;
TreeNode left = null;
TreeNode right = null;
public TreeNode(int val) {
this.val = val;
}
}
*/
public class Solution {
public ArrayList<Integer> PrintFromTopToBottom(TreeNode root) {
if(root == null) return new ArrayList<Integer>();
ArrayList<Integer> result = new ArrayList<Integer>();
// List.remove(0)相当于出队列
ArrayList<TreeNode> queue = new ArrayList<TreeNode>();
queue.add(root);
// 当queue不为空时
while(queue.size()>0){
TreeNode temp = queue.remove(0);
result.add(temp.val);
if(temp.left != null) {
queue.add(temp.left);
}
if(temp.right != null) {
queue.add(temp.right);
}
}
return result;
}
}
总结:
1.二叉树层遍历,首先想到使用队列或者栈
每层都是从右往左 -> 队列
s型遍历 -> 两个栈
2.list.remove(0)可以用来实现队列的出队