ツリーを上から下へ印刷(java版)


【タイトル説明】ツリーの各ノードを上から下へ印刷し、同層ノードを左から右へ印刷します.
【解題の考え方】//1.二叉樹を層順に遍歴する.//2.ArrayListを使用して、順次アクセスしたノードを記録します.
import java.util.ArrayList;
import java.util.LinkedList;
import java.util.Queue;

public class Solution {
    public ArrayList<Integer> PrintFromTopToBottom(TreeNode root) {
        ArrayList<Integer> arr = new ArrayList<Integer>();
        Queue<TreeNode> q = new LinkedList<TreeNode>();
        if(root != null){
            q.offer(root);
        }
        while(!q.isEmpty()){
            TreeNode tr = q.poll();
            arr.add(tr.val);
            System.out.println(tr.val);
            if(tr.left != null){
                q.offer(tr.left);
            }
            if(tr.right != null){
                q.offer(tr.right);
            }
        }
        return arr;
    }
}