LeetCode(107): 二叉樹的層次遍歷 II


Easy!

題目描述:

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

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

    3
   / \
  9  20
    /  \
   15   7

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

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

解題思路:

從底部層序遍歷其實還是從頂部開始遍歷,只不過最后存儲的方式有所改變,參見http://www.cnblogs.com/grandyang/p/4051321.html,代碼如下:

C++解法一:

 1 // Iterative
 2 class Solution {
 3 public:
 4     vector<vector<int> > levelOrderBottom(TreeNode *root) {
 5         vector<vector<int> > res;
 6         if (root == NULL) return res;
 7 
 8         queue<TreeNode*> q;
 9         q.push(root);
10         while (!q.empty()) {
11             vector<int> oneLevel;
12             int size = q.size();
13             for (int i = 0; i < size; ++i) {
14                 TreeNode *node = q.front();
15                 q.pop();
16                 oneLevel.push_back(node->val);
17                 if (node->left) q.push(node->left);
18                 if (node->right) q.push(node->right);
19             }
20             res.insert(res.begin(), oneLevel);
21         }
22         return res;
23     }
24 };

下面我們來看遞歸的解法,核心就在於我們需要一個二維數組,和一個變量level,當level遞歸到上一層的個數,我們新建一個空層,繼續往里面加數字。

C++解法二:

 1 // Recurive
 2 class Solution {
 3 public:
 4     vector<vector<int>> levelOrderBottom(TreeNode* root) {
 5         vector<vector<int> > res;
 6         levelorder(root, 0, res);
 7         return vector<vector<int> > (res.rbegin(), res.rend());
 8     }
 9     void levelorder(TreeNode *root, int level, vector<vector<int> > &res) {
10         if (!root) return;
11         if (res.size() == level) res.push_back({});
12         res[level].push_back(root->val);
13         if (root->left) levelorder(root->left, level + 1, res);
14         if (root->right) levelorder(root->right, level + 1, res);
15     }
16 };

 


免責聲明!

本站轉載的文章為個人學習借鑒使用,本站對版權不負任何法律責任。如果侵犯了您的隱私權益,請聯系本站郵箱yoyou2525@163.com刪除。



 
粵ICP備18138465號   © 2018-2025 CODEPRJ.COM