劍指offer22-上往下打印二叉樹
最近全國疫情嚴重,待在家裏沒事幹,馬上又要準備春招了,最近刷刷題,記錄一下!再說一句,武漢加油,大家出門記得戴口罩!
1、題目描述
從上往下打印出二叉樹的每個節點,同層節點從左至右打印。
2、思路
本題實際上就是二叉樹的層次遍歷,深度遍歷可以用遞歸或者棧,而層次遍歷很明顯應該使用隊列。同樣我們可以通過一個例子來分析得到規律:每次打印一個結點時,如果該結點有子結點,則將子結點放到隊列的末尾,接下來取出隊列的頭重複前面的打印動作,直到隊列中所有的結點都打印完畢。
3、代碼
public class Solution {
public ArrayList<Integer> PrintFromTopToBottom(TreeNode root) {
//保存結果
ArrayList<Integer> res=new ArrayList<>();
if(root==null)
return res;
Queue<TreeNode> queue=new LinkedList<>(); //定義一個隊列
queue.add(root);
while(queue.size()!=0)
{
TreeNode temp=queue.poll();//隊頭移除
res.add(temp.val);
if(temp.left!=null)
queue.add(temp.left);
if(temp.right!=null)
queue.add(temp.right);
}
return res;
}
}