Binary Tree Level Order Traversal II

Binary Tree Level Order Traversal II

題目描述

Given a binary tree, return the bottom-up level order traversal of its nodes’ values. (ie, from left to right, level by level from leaf to root).

For example:
Given binary tree {3,9,20,#,#,15,7},

    3
   / \
  9  20
    /  \
   15   7

return its bottom-up level order traversal as:

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

解題思路

首先可以去看一下對LevelOrderTraversal的分析，我在前面的文章中有介紹，此題也是層次遍歷，只是要求返回的結果集結構不同，結果集中在前的是底層的節點集合，也就是從下向上從左向右這樣遍歷，我們采用和LevelOrderTraversal1中一樣的結構，記錄層次信息，采用同樣的輔助隊列去遍歷，和1不同的就是我們遍歷完一個層次後在加入結果集的時候是插入在頭部，這樣結果集的頭部就是底層的節點結合，這裡我們使用LinkedList作為List接口的實現類，這樣在插入數據的時候效率高，O(1),那麼代碼和1不同的就是我們使用LinkedList取代了ArrayList，代碼中有兩處add函數，改為add(0,List)即可。

詳細代碼

/**
 * Definition for a binary tree node.
 * public class TreeNode {
 *     int val;
 *     TreeNode left;
 *     TreeNode right;
 *     TreeNode(int x) { val = x; }
 * }
 */
    //Binary Tree Level Order Traversal
    class TreeNodeWithDepth
    {
        TreeNode treeNode;
        int depth;

        public TreeNodeWithDepth(TreeNode treeNode, int depth)
        {
            this.treeNode = treeNode;
            this.depth = depth;
        }
    }

public class Solution {
    //Given a binary tree, return the level order traversal of its nodes' values. (ie, from left to right, level by level).
    //二叉樹的層次遍歷 2 最底層最先遍歷
    public List> levelOrderBottom(TreeNode root) {
        int depth = 1;
        int lastLevel = 1; //while中用來標示上一次節點的level
        List> resultsList = new LinkedList>();  //注意初始化 List是接口類 //這點和levelOrder不同使用LinkedList
        Queue queue = new LinkedList<>();
        TreeNodeWithDepth pNode = null;
        if(root != null)
        {
            pNode = new TreeNodeWithDepth(root, 1);
            queue.offer(pNode);
        }
        List listPerLevel = new ArrayList<>();//每層的list 
        while(!queue.isEmpty())
        {
            pNode = queue.poll();//程序加入時已經判斷不為空節點

            if(pNode.depth > lastLevel)
            {
                //說明是新的一層
                if(listPerLevel.size()!=0)
                {
                    //把上一層的列表加入result中 插入到第一個位置 
                    resultsList.add(listPerLevel);
                }
                listPerLevel = new ArrayList<>(); //重新另起一個listPerLevel
                listPerLevel.add(pNode.treeNode.val);//在新列表中加入當前節點
                lastLevel = pNode.depth;//更新lastLevel
            }
            else 
            {
                listPerLevel.add(pNode.treeNode.val);
            }
            //確定子節點深度
            depth = pNode.depth+1;

            //子節點入隊列
            if(pNode.treeNode.left != null)
            {
                queue.offer(new TreeNodeWithDepth(pNode.treeNode.left, depth));
            }
            if(pNode.treeNode.right != null)
            {
                queue.offer(new TreeNodeWithDepth(pNode.treeNode.right, depth));
            }

        }

        //while循環中最後一次的listPerLevel也要加入
        if(listPerLevel.size()>0)
        {
            resultsList.add(0,listPerLevel);
        }
        return resultsList;
    }
}