剑指 Offer 32 - I. 从上到下打印二叉树

最新推荐文章于 2024-09-21 10:02:17 发布

zhazha_hui

最新推荐文章于 2024-09-21 10:02:17 发布

阅读量68

点赞数

分类专栏： leetcode 文章标签：二叉树广度优先遍历层次遍历数据结构算法

本文链接：https://blog.csdn.net/zhazha_hui/article/details/113029076

版权

leetcode 专栏收录该内容

49 篇文章 0 订阅

订阅专栏

一枪秒了，有什么好说的。这种要按照层完成操作的，一般都需要用到广度优先遍历。

class Solution {
    public int[] levelOrder(TreeNode root) {
        if(root == null)
            return new int[0];
        Queue<TreeNode> print = new LinkedList();
        ArrayList<Integer> res = new ArrayList();
        print.add(root);
        while(!print.isEmpty()){
            TreeNode head = print.poll();
            res.add(head.val);
            if(head.left != null)
                print.add(head.left);
            if(head.right != null)
                print.add(head.right);
        }
        int[] ans = new int[res.size()];
        for(int i = 0; i < ans.length; i++)
            ans[i] = res.get(i);
        return ans;
    }
}