题目
从上到下按层打印二叉树,同一层结点从左至右输出。每一层输出一行。
思想
队列
代码
import java.util.ArrayList;
import java.util.LinkedList;
/*
public class TreeNode {
int val = 0;
TreeNode left = null;
TreeNode right = null;
public TreeNode(int val) {
this.val = val;
}
}
*/
public class Solution {
ArrayList<ArrayList<Integer>> Print(TreeNode pRoot) {
ArrayList<ArrayList<Integer>> result = new ArrayList<ArrayList<Integer>>();
if(pRoot == null)
return result;
LinkedList<TreeNode> que = new LinkedList<TreeNode>();
que.add(pRoot);
while(!que.isEmpty())
{
int rowSize = que.size();
ArrayList<Integer> row = new ArrayList<Integer>();
for(int i = 0;i < rowSize;i++)
{
TreeNode head = que.remove();
row.add(head.val);
if(head.left != null)
que.add(head.left);
if(head.right != null)
que.add(head.right);
}
result.add(new ArrayList<Integer>(row));
}
return result;
}
}