题目:从上到下按层打印二叉树,同一层结点从左至右输出。每一层输出一行。
- 建立一个Queue< TreeNode60>,为存放每一层节点做准备
- 建立一个ArrayList< Integer> ,按从左到右存放每一层节点
- 建立一个ArrayList<ArrayList< Integer>>,按从上到下存放每一层
import java.util.ArrayList;
import java.util.LinkedList;
import java.util.Queue;
public class JZ60 {
ArrayList<ArrayList<Integer> > Print(TreeNode60 pRoot) {
ArrayList<ArrayList<Integer>> list = new ArrayList<>();
ArrayList<Integer> tmp = new ArrayList<>();//临时存放每一层的节点
Queue<TreeNode60> node = new LinkedList<TreeNode60>();
if (pRoot == null) { return list; }
node.add(pRoot);//把根节点加入到队列中
int start = 0;//起始指针
int end = 1;//结尾指针,这里因为加入了根节点,所以初始值是1
while(!node.isEmpty()){
TreeNode60 t=node.poll();
tmp.add(t.val);//添加到临时存放list
start++;//起始指针后移
if (t.left!=null){//将子节点添加到队列
node.add(t.left);
}
if (t.right!=null){//将子节点添加到队列
node.add(t.right);
}
if (start==end){//判断本层是不是完成
start=0;//置0
end=node.size();//这里设计很妙!赞赞赞
list.add(new ArrayList<>(tmp));//添加到list
tmp.clear();//清空,为下一层做准备
}
}
return list;
}
}
class TreeNode60 {
int val = 0;
TreeNode60 left = null;
TreeNode60 right = null;
public TreeNode60(int val) {
this.val = val;
}
}
小结:真佩服前人,任重而道远啊!加油,高同学!