107. 二叉树的层序遍历 II

小豆丁 1年前 ⋅ 948 阅读
107. 二叉树的层序遍历 II
给定一个二叉树,返回其节点值自底向上的层序遍历。 (即按从叶子节点所在层到根节点所在的层,逐层从左向右遍历)

例如:
给定二叉树 [3,9,20,null,null,15,7],

    3
   / \
  9  20
    /  \
   15   7
返回其自底向上的层序遍历为:

[
  [15,7],
  [9,20],
  [3]
]

解析

和层序遍历一样的,倒着插入就好了

/**
 * Definition for a binary tree node.
 * public class TreeNode {
 *     int val;
 *     TreeNode left;
 *     TreeNode right;
 *     TreeNode(int x) { val = x; }
 * }
 */
class Solution {
    public List<List<Integer>> levelOrderBottom(TreeNode root) {
        LinkedList<List<Integer>> res = new LinkedList<>();
        if(root==null) return res;
        LinkedList<TreeNode> queue = new LinkedList<>();
        TreeNode head = root;
        queue.add(head);
        while(!queue.isEmpty()){
            int size = queue.size();
            List<Integer> tmp = new ArrayList<>();
            for(int i = 0 ;i <size;i++){
                head = queue.removeFirst();
                if(head.left!=null) queue.add(head.left);
                if(head.right!=null) queue.add(head.right);
                tmp.add(head.val);
            }
            res.addFirst(tmp);
        }
        return res;
    }
}