- 题目描述:
从上到下按层打印二叉树,同一层结点从左至右输出。每一层输出一行。 - 分析:
利用层次遍历的算法,设置变量last指向当前层的最后一个节点,设置变量count记录当前层已经访问的节点的个数,当count等于last时,表示该层访问结束。
层次遍历在求树的宽度、输出某一层节点,某一层节点个数,每一层节点个数都可以采取类似的算法。
树的宽度:在树的深度算法基础上,加一个记录访问过的层节点个数最多的变量max,在访问每层前max与last比较,如果max比较大,max不变,如果max小于last,把last赋值给max.
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>> list1 = new ArrayList<ArrayList<Integer>>();
LinkedList<TreeNode> queue = new LinkedList<TreeNode>();
if(pRoot == null)
return list1;
TreeNode current = pRoot;
queue.offer(current);
int count; //记录当前每层结点已经打印的个数
int last; //记录每层最后结点的位置
while(!queue.isEmpty()){
count = 0;
last = queue.size();
ArrayList<Integer> list2 = new ArrayList<Integer>();
while(count < last){
current = queue.pop();
list2.add(current.val);
count++;
if(current.left != null){
queue.offer(current.left);
}
if(current.right != null){
queue.offer(current.right);
}
}
list1.add(list2);
}
return list1;
}
}