题目:
有一棵二叉树,请设计一个算法,按照层次打印这棵二叉树。
给定二叉树的根结点root,请返回打印结果,结果按照每一层一个数组进行储存,所有数组的顺序按照层数从上往下,且每一层的数组内元素按照从左往右排列。保证结点数小于等于500。
解析:
这里的关键数据结构是使用队列来存储节点,因为队列是先进先出顺序,可以保证节点按照顺序弹出,问题的难点是如何换行打印,可以使用两个变量 last和nlast,一个指向当前打印的最后一个节点,一个指向下一行的最后一个节点,这样难点转换成了last和nlast的更新。nlast设置为进入队列的最新一个节点,因为最新一个节点就能保证是下一行的最后节点,last更新是当换行时,last=nlast。换行的标志是打印的当前节点是最后一个节点,假设从队列中弹出的节点为temp,即temp==last时换行。
java代码实现(该实现已经在牛客网通过):
import java.util.*;
public class TreeNode {
int val = 0;
TreeNode left = null;
TreeNode right = null;
public TreeNode(int val) {
this.val = val;
}
}
public class TreePrinter {
public int[][] printTree(TreeNode root) {
// write code here
if(root == null)
return null;
ArrayList<ArrayList<Integer>> arrys = new ArrayList<>();
ArrayList<Integer> line = new ArrayList<>();
Queue<TreeNode> queue = new LinkedList<TreeNode>();
queue.offer(root);
TreeNode last = root;
TreeNode nLast = root;
TreeNode temp = null;
while(!queue.isEmpty()){
temp = queue.poll();
line.add(temp.val);
if(temp.left!=null){
nLast = temp.left;
queue.offer(temp.left);
}
if(temp.right!=null){
nLast = temp.right;
queue.offer(temp.right);
}
if(temp == last){ // 边界这里容易出错
last = nLast;
arrys.add(new ArrayList<Integer>(line));// 这里特别容易出错,直接add(line)是不行的
line.clear();
}
}
int rows = arrys.size();
int result[][] = new int[rows][];
for(int i=0;i<rows;i++){
int columns = arrys.get(i).size();
result[i] = new int[columns];
for(int j=0;j<columns;j++){
result[i][j] = arrys.get(i).get(j);
}
}
return result;
}
}