(Java）leetcode-107 Binary Tree Level Order Traversal II（二叉树的层次遍历 II ）

Mr.Bean-Pig

于 2020-06-09 15:33:25 发布

阅读量85

点赞数

分类专栏：算法题解文章标签：二叉树 bfs 队列 stack queue

本文链接：https://blog.csdn.net/z714405489/article/details/106643478

版权

算法题解专栏收录该内容

140 篇文章 0 订阅

订阅专栏

题目描述

给定一个二叉树，返回其节点值自底向上的层次遍历。（即按从叶子节点所在层到根节点所在的层，逐层从左向右遍历）
在这里插入图片描述

思路

层次遍历，可以直接套BFS框架：

        Queue<TreeNode> queue = new LinkedList<>();
        queue.offer(root);
        while(!queue.isEmpty()) {
            int size = queue.size();

            for (int i = 0; i < size; i++) {
                TreeNode current = queue.poll();
                if (current.left != null) queue.offer(current.left);
                if (current.right != null) queue.offer(current.right);                
            }
        }

需要注意的就是结果与从上往下的顺序相反，需要对正向结果做一次倒置，那么用栈就可以实现啦。

代码

class Solution {
    public List<List<Integer>> levelOrderBottom(TreeNode root) {
    	List<List<Integer>> resultList = new ArrayList<>();
    	if (root == null) return resultList;

    	Stack<List<Integer>> stack = new Stack<>();
        Queue<TreeNode> queue = new LinkedList<>();

        queue.offer(root);
        while(!queue.isEmpty()) {
            int size = queue.size();
            List<Integer> levelList = new ArrayList<>();
            for (int i = 0; i < size; i++) {
                TreeNode current = queue.poll();
                levelList.add(current.val);
                if (current.left != null) queue.offer(current.left);
                if (current.right != null) queue.offer(current.right);                
            }
            stack.push(levelList);
        }       
        while(!stack.empty()) resultList.add(stack.pop());
        return resultList;
    }
}