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);
    }