1. 程式人生 > 其它 >107. 二叉樹的層次遍歷 II

107. 二叉樹的層次遍歷 II

難度 medium
給定一個二叉樹,返回其節點值自底向上的層次遍歷。 (即按從葉子節點所在層到根節點所在的層,逐層從左向右遍歷)

例如:
給定二叉樹 [3,9,20,null,null,15,7],

    3
   / \
  9  20
    /  \
   15   7

返回其自底向上的層次遍歷為:

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

解題思路:用單向佇列來層序遍歷二叉樹是通用的做法,這裡因為是從下到上遍歷,每次將一層的資料儲存在一個一維陣列中,插入二維陣列的時候保證插入的位置是第0位即可,這樣就能保證結果是逆序的。

程式碼 t100 s14 java

class Solution {
    public List<List<Integer>> levelOrderBottom(TreeNode root) {
        List<List<Integer>> res = new ArrayList<>();
        Queue<TreeNode> pq = new LinkedList<>();
        if(root==null) return res;
        pq.add(root);
        while(!pq.isEmpty()){
            int sz = pq.size();
            List<Integer> t = new ArrayList<>();
            for(int i=0; i<sz; i++){
                TreeNode node = pq.poll();
                t.add(node.val);
                if(node.left!=null) pq.add(node.left);
                if(node.right!=null) pq.add(node.right);
            }
            res.add(0, t);
        }
        return res;
    }
}

下面這個是4個月前的cpp版本,用一個層次遍歷,先按從上往下的順序把每一層push進一個vector< vector< int>>裡面,然後再反轉一下。
程式碼t53 s8 cpp

class Solution {
public:
    vector<vector<int>> levelOrderBottom(TreeNode* root) {
        vector<vector<int>> res;
        if(!root) return res;
        queue<TreeNode*> q;
        q.push(root);
        while(!q.empty()){
            int t = q.size();
            vector<int> vec;
            for(int i=0; i<t; i++){
                TreeNode* temp = q.front();
                q.pop();
                vec.push_back(temp->val);
                if(temp->left) q.push(temp->left);
                if(temp->right) q.push(temp->right);
            }
            res.push_back(vec);
        }
        reverse(res.begin(), res.end());        
        return res;
    }
};

參考資料