LeetCode-107.二叉树的层次遍历II(相关话题:广度优先/栈)

给定一个二叉树,返回其节点值自底向上的层次遍历。 (即按从叶子节点所在层到根节点所在的层,逐层从左向右遍历)

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

    3
   / \
  9  20
    /  \
   15   7

返回其自底向上的层次遍历为:

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

Java代码:

class Solution {
    public List<List<Integer>> levelOrderBottom(TreeNode root) {
        Stack<List<Integer>> s = new Stack<>();
        if(null != root){
            List<List<TreeNode>> lists = new LinkedList<>();
            List<TreeNode> l0 = new ArrayList<>();
            l0.add(root);
            lists.add(l0);
            for(int i = 0; i < lists.size(); i++){
                List<Integer> si = new LinkedList<>();
                List<TreeNode> curList = lists.get(i);
                for(int j = 0; j < curList.size(); j++){
                    TreeNode cur = curList.get(j);
                    si.add(cur.val);
                    if(null != cur.left || null != cur.right){
                        if(lists.size()-1 == i){
                            List<TreeNode> li = new LinkedList<>();
                            lists.add(li);
                        }
                        if(null != cur.left)
                            lists.get(i+1).add(cur.left);
                        if(null != cur.right)
                            lists.get(i+1).add(cur.right);
                    }
                }
                s.push(si);
            }
        }
        List<List<Integer>> res = new ArrayList<>(s.size());
        for(int i = s.size()-1; i >= 0; i--)
            res.add(s.pop());
        return res;
    }
}

猜你喜欢

转载自blog.csdn.net/weixin_38823568/article/details/82982779
今日推荐