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

最新推荐文章于 2022-03-22 13:59:24 发布

lankerens

最新推荐文章于 2022-03-22 13:59:24 发布

阅读量60

点赞数

分类专栏：剑指Offer 文章标签：队列 queue 二叉树

本文链接：https://blog.csdn.net/qq_43765535/article/details/108738179

版权

剑指Offer 专栏收录该内容

60 篇文章 0 订阅

订阅专栏

LeetCode：剑指 Offer 32 - III. 从上到下打印二叉树 III

在这里插入图片描述

树的层序遍历 >> 结合队列 Queue<TreeNode> queue = new LinkedList<>()

正常的将节点从左到右添加进队列，在特定的层数中，将 list 反转即可


    List<List<Integer>> ans = new ArrayList<>();
    List<Integer> list = new ArrayList<>();

    public List<List<Integer>> levelOrder(TreeNode root) {
        if (root == null) {
            return ans;
        }

        dfs(root);

        return ans;
    }

	// 记录层数 
    int cnt = 0;

    public void dfs(TreeNode node) {
        Queue<TreeNode> queue = new LinkedList<>();
        queue.offer(node);
        while (!queue.isEmpty()) {
        	// size 记录上一层节点的个数
            int size = queue.size();
            for (int i = 0; i < size; i++) {
                // 反向输出
                TreeNode pop = ((LinkedList<TreeNode>) queue).pop();

                if (pop.left != null) {
                    queue.offer(pop.left);
                }
                if (pop.right != null) {
                    queue.offer(pop.right);
                }

                // 先统一从左到右add, 后面 reverse
                list.add(pop.val);
            }

            if(cnt % 2 == 1){
                // 奇数 ---- 右到左
                Collections.reverse(list);
            }
            ans.add(new ArrayList<>(list));
            list.clear();
            cnt++;
        }

    }