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;
}
}