剑指offer | 从上往下打印二叉树

从上往下打印二叉树

描述

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

代码1

public class Solution {
    public ArrayList<Integer> PrintFromTopToBottom(TreeNode root) {
        ArrayList<Integer> result = new ArrayList<>();
        if (root == null)
            return result;
        ArrayList<TreeNode> queue = new ArrayList<>();
        queue.add(root);
        while (!queue.isEmpty()) {
            TreeNode curNode = queue.remove(0);
            result.add(curNode.val);
            if (curNode.left != null)
                queue.add(curNode.left);
            if (curNode.right != null)
                queue.add(curNode.right);
        }
        return result;
    }
}

代码2

public class Solution {
    public ArrayList<Integer> PrintFromTopToBottom(TreeNode root) {
        ArrayList<Integer> result = new ArrayList<>();
        if (root == null)
            return result;
        ArrayList<TreeNode> queue = new ArrayList<>();
        queue.add(root);
        int count = 0;
        while (count < queue.size()) {
            // TreeNode curNode = queue.remove(0);  // 这是仿照队列的出队列方式，但删除比较耗时间，直接用索引获取即可
            TreeNode curNode = queue.get(count);
            ++count;
            result.add(curNode.val);
            if (curNode.left != null)
                queue.add(curNode.left);
            if (curNode.right != null)
                queue.add(curNode.right);
        }
        return result;
    }
}

思路

• 上面两段代码都可以，稍微有一些不同；
• 本题是二叉树的层次遍历，这种非递归（借助队列或者类似的数据结构）的方式也可以用来实现其他二叉树的问题，比如二叉树的镜像，也可以用上面非递归的思路来做。