AcWing打卡活动
《剑指Offer》打卡活动
周三第十题 分行从上往下打印二叉树
/**
* Definition for a binary tree node.
* public class TreeNode {
* int val;
* TreeNode left;
* TreeNode right;
* TreeNode(int x) { val = x; }
* }
* 思路:
* 和上题差不多
* 要求多了,需要换行,打印层级的时候,需要使用变量将这一层打印的过程记录下来
*
*
*/
class Solution {
public List<List<Integer>> printFromTopToBottom(TreeNode root) {
List<List<Integer>> printList = new ArrayList();
List<Integer> elementList = new ArrayList();
// 当前层中,每有一个元素,next++
int next = 0;
// 记录当前层的元素是否打印完,树的第一层只有一个根节点,所以只会打印一个元素
int toBePrintln = 1;
Queue<TreeNode> nodeQueue = new LinkedList();
TreeNode node;
nodeQueue.offer(root);
while(nodeQueue.peek() != null) {
node = nodeQueue.poll();
elementList.add(node.val);
if(node.left != null) {
nodeQueue.offer(node.left);
next++;
}
if(node.right != null) {
nodeQueue.offer(node.right);
next++;
}
toBePrintln--;
if(toBePrintln == 0) {
printList.add(elementList);
elementList = new ArrayList();
// 赋值下一层的元素数
toBePrintln = next;
next = 0;
}
}
return printList;
}
}