1. 程式人生 > >從上往下打印二叉樹

從上往下打印二叉樹

ott ger pub AD 二叉樹的層次遍歷 結構 pty bottom tree

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

給定的樹節點結構:

public class TreeNode {
    int val = 0;
    TreeNode left = null;
    TreeNode right = null;
 
    public TreeNode(int val) {
        this.val = val;
    }
 
}

二叉樹的層次遍歷,借助一個隊列來實現。

代碼如下:

public class Solution {
    public ArrayList<Integer> PrintFromTopToBottom(TreeNode root) {
        ArrayList<Integer> arrayList = new ArrayList<>();
        Queue<TreeNode> queue = new ArrayDeque<>();
 
        if (root != null) queue.offer(root);
 
        while (!queue.isEmpty()) {
            TreeNode node = queue.poll();
            arrayList.add(node.val);
            if (node.left != null) queue.offer(node.left);
            if (node.right != null) queue.offer(node.right);
        }
 
        return arrayList;
    }
}

從上往下打印二叉樹