天天看點

從上往下列印出二叉樹的每個節點,同層節點從左至右列印

題目描述

從上往下列印出二叉樹的每個節點,同層節點從左至右列印。

import java.util.ArrayList;
/**
public class TreeNode {
    int val = 0;
    TreeNode left = null;
    TreeNode right = null;

    public TreeNode(int val) {
        this.val = val;

    }

}
     思路:層次周遊,用arraylist模拟一個隊列來存儲相應的TreeNode
*/
import java.util.LinkedList;
public class Solution {
    public ArrayList<Integer> PrintFromTopToBottom(TreeNode root) {
        //注意不能直接傳回null,不然樹為空時,不能通過
        if(root ==null) return new ArrayList<Integer>();
        //1、定義一個Arraylist對象來存儲二叉樹的節點值
        ArrayList<Integer> arr = new ArrayList<>();
        //2、用linkedList來做層次周遊
        LinkedList<TreeNode> list = new LinkedList<>();
        list.add(root);
        while(!list.isEmpty()){
            TreeNode temp = list.removeFirst();
            if(temp.left!=null){
                list.add(temp.left);
            }
            if(temp.right!=null){
                list.add(temp.right);
            }
            arr.add(temp.val);
        }
        return arr;
         
    }
}