题解 | #把二叉树打印成多行#
把二叉树打印成多行
http://www.nowcoder.com/practice/445c44d982d04483b04a54f298796288
import java.util.*;
/*
public class TreeNode {
int val = 0;
TreeNode left = null;
TreeNode right = null;
public TreeNode(int val) {
this.val = val;
}
}
*/
这个题是之字形打印二叉树的一部分,也就是层序遍历二叉树
相对还是比较简单
总的思路是使用队列,维护的方式是取出队首并放入它的左右孩子
然后利用双重循环,每一次外层循环代表遍历完一层的节点,每一次内层循环代表遍历一个节点
外层维护一个用来装每一次节点VAL值的集合,以及一个用来统计下一层节点数的temp
内层执行取出队首并放入其左右子节点的操作,每取出一个节点temp加1.
内层结束把temp赋给sum,然后把row赋给总的集合
最后总循环结束,所以层数都被遍历完成,返回result总集合即可。
public class Solution {
ArrayList<ArrayList<Integer> > Print(TreeNode pRoot) {
ArrayList<ArrayList<Integer>> result = new ArrayList<ArrayList<Integer>>();
if(pRoot == null)return result;//虽然是没什么必要,但是还是要判断一下数组越界
Queue<TreeNode> queue = new LinkedList<TreeNode>();
int sum = 1;//用来统计每层的个数
queue.add(pRoot);
while(!queue.isEmpty()){//注意==是数组判断,isEmpty是内容判断,所以判断引用类型用isEmpty
int temp = 0;//用来存放下一层的节点个数
ArrayList<Integer> row = new ArrayList<Integer>();
while(sum > 0){
TreeNode x = queue.poll();//取出队首节点
row.add(x.val);//然后把它的值加入到行集合中
sum--;
if(x.left != null){
queue.add(x.left);
temp++;
}
if(x.right != null){
queue.add(x.right);
temp++;
}
}
sum = temp;//要在整个一层搞定了再把temp赋值给sum。
result.add(row);
}
return result;
}
}