从上往下打印出二叉树的每个节点,同层节点从左至右打印。
看到题目分两种情况:
1.如果是空树,该怎么返回值?
2.如果不是空树,要怎么按层遍历?
按照题目描述:从上往下,从左往右,很显然这个描述的先入先出,故想到队列,但题目要求输出的是ArrayList。故先让结点入队列,将出栈结点的值入ArrayList中.。记得将出栈的结点保存到临时变量,让问题规模依次减小。
//记得导包进来!!!
import java.util.ArrayList;
import java.util.Deque;
import java.util.LinkedList;
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) {
ArrayList<Integer> list = new ArrayList<Integer>();
Deque<TreeNode> queue = new LinkedList<TreeNode>();
if(root == null)
return list;
queue.add(root);
while(!queue.isEmpty()){
//重新定义个结点保存出栈的结点
TreeNode t= queue.poll();
list.add(t.val);
/**
* 临时结点 t在while循环中起了很大作用,它才是将循环
* 结束的大功臣哟。o(* ̄▽ ̄*)ブ
*/
if(t.left != null){
queue.add(t.left);
}
if(t.right != null){
queue.add(t.right);
}
}
return list;
}
}