107. Binary Tree Level Order Traversal II

一、题目

  1、审题

  2、分析

    给出一个二叉树,从下往上输出每一层的节点值。

二、解答

  1、思路:

    方法一、BFS

      利用一个队列进行层次遍历,同时 List 进行插入时,利用 add(index, val) 方法, index 指定插入的 List 的下标。

public List<List<Integer>> levelOrderBottom(TreeNode root) {

            
        List<List<Integer>> resultList = new LinkedList<>();
        if(root == null)
            return resultList;
        
        Queue<TreeNode> queue = new LinkedList<>();
        queue.add(root);
        while(!queue.isEmpty()) {
            List<Integer> list = new LinkedList<>();
            int size = queue.size();
            for (int i = 0; i < size; i++) {
                TreeNode node = queue.poll();
                list.add(node.val);
                if(node.left != null)
                    queue.add(node.left);
                if(node.right != null)
                    queue.add(node.right);
            }
            resultList.add(0, list);
        }
        
        return resultList;
    }

  方法二、DFS

    采用递归进行层次遍历,从上往下存于一个 List 中,最后对 List 进行翻转。

public List<List<Integer>> levelOrderBottom2(TreeNode root) {
        
        List<List<Integer>> resultList = new ArrayList<List<Integer>>();
        helperLevelOrderBottom(root, 0, resultList);
        
        List<List<Integer>> tmpList = new LinkedList<>();
        for (int i = resultList.size() - 1; i >= 0; i--) 
            tmpList.add(resultList.get(i));
        
        return tmpList;
    }
    
    private void helperLevelOrderBottom(TreeNode root, int height,
            List<List<Integer>> resultList) {
        
        if(root == null)
            return;
        
        if(resultList.size() <= height)
            resultList.add(new ArrayList<Integer>());
        resultList.get(height).add(root.val);
        
        helperLevelOrderBottom(root.left, height+1, resultList);
        helperLevelOrderBottom(root.right, height+1, resultList);
        
    }

  优化,调用 helper 方法时,直接将节点值插入对应的从下向上的 List 的对应的位置。

public List<List<Integer>> levelOrderBottom3(TreeNode root) {
        
        List<List<Integer>> resultList = new ArrayList<List<Integer>>();
        helperLevelOrderBottom2(root, 0, resultList);
        
        return resultList;
    }
    
    private void helperLevelOrderBottom2(TreeNode root, int level, List<List<Integer>> resultList) {
        
        if(root == null)
            return;
        
        if(resultList.size() <= level)    // 在 List 头插入
            resultList.add(0, new ArrayList<Integer>());
        
//        resultList.get(0).add(root.val);    Wrong, is DFS
        resultList.get(resultList.size() - level - 1).add(root.val);
        helperLevelOrderBottom2(root.left, level + 1, resultList);
        helperLevelOrderBottom2(root.right, level + 1, resultList);
    }
原文地址:https://www.cnblogs.com/skillking/p/9734307.html