《剑指Offer》32.1 从上往下打印二叉树

题目链接

牛客网

题目描述

从上往下打印出二叉树的每个节点，同层节点从左至右打印。

例如，以下二叉树层次遍历的结果为：1,2,3,4,5,6,7

解题思路

public class Solution {
    public ArrayList<Integer> PrintFromTopToBottom(TreeNode root) {
        ArrayList<Integer> res = new ArrayList<>();
        if (root==null) return res;
        Queue<TreeNode> q = new LinkedList<>();
        q.add(root);
        while (!q.isEmpty()) {
            int cnt = q.size();
            while (cnt-- > 0) {
                TreeNode node = q.remove();
                if (node==null) continue;
                q.add(node.left);
                q.add(node.right);
                res.add(node.val);
            }
        }
        return res;
    }
}