LeetCode 107. Binary Tree Level Order Traversal II

最新推荐文章于 2020-01-03 13:57:10 发布

zjxxyz123

最新推荐文章于 2020-01-03 13:57:10 发布

阅读量116

点赞数

分类专栏： LeetCode 文章标签： LeetCode 二叉树 BFS DFS

本文链接：https://blog.csdn.net/zjxxyz123/article/details/80021895

版权

LeetCode 专栏收录该内容

133 篇文章 1 订阅

订阅专栏

问题描述

Given a binary tree, return the bottom-up level order traversal of its nodes’ values. (ie, from left to right, level by level from leaf to root).
For example:
地址

问题分析

该题是 LeetCode 102. Binary Tree Level Order Traversal 的进阶，只不过该题要求序列是倒序的。
BFS
- 同样是像原题一样层次遍历，只不过需要对结果进行逆序。有以下几种方法实现
  - 对最终的结果逆序，利用Collections.reverse(res)
  - 直接将新形成的列表插入首部，res.add(0, levelList)
  - 利用双端队列的addFirst()
DFS
- 根据传入的level确定当前节点应该串在那一条链表中

代码实现

    public List<List<Integer>> levelOrderBottom(TreeNode root) {
        if (root == null) {
            return new ArrayList<>();
        }
        LinkedList<TreeNode> queue = new LinkedList<>();
        List<List<Integer>> res = new ArrayList<>();
        queue.add(root);
        while (! queue.isEmpty()) {
            int levelSize = queue.size();
            List<Integer> levelList = new ArrayList<>();
            for (int i = 0; i < levelSize; i++) {
                TreeNode popNode = queue.remove();
                levelList.add(popNode.val);
                if (popNode.left != null) {
                    queue.add(popNode.left);
                }
                if (popNode.right != null) {
                    queue.add(popNode.right);
                }
            }
            //每次存在下标为0的位置，比最后进行反转更快！
            res.add(0, levelList);
        }
       // Collections.reverse(res);
        return res;
    }

    public List<List<Integer>> levelOrderBottom(TreeNode root) {
        if (root == null) {
            return new ArrayList<>();
        }
        LinkedList<List<Integer>> res = new LinkedList<List<Integer>>();
        //初始dfs
        dfs(root, 0, res);
        return res;
    }

    public void dfs(TreeNode root, int level,  LinkedList<List<Integer>> res) {
        if (root == null) {
            return;
        }
        if (res.size() < level + 1) { //如果还没有这一层的list，则new一个，并放入首部
            //res.add(0, new ArrayList<>());
            res.addFirst(new ArrayList<>());
        }
        //将该节点存入相应的list
        res.get(res.size() - level - 1).add(root.val);
        //进行左右孩子的 dfs
        dfs(root.left, level + 1, res);
        dfs(root.right, level + 1, res);
    }