指offer22-上往下打印二叉樹

劍指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;
    }
}
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章