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

leetcode------107. 二叉樹的層次遍歷 II

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

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

    3
   / \
  9  20
    /  \
   15   7

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

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

------------------------------------------------------------------------------------------------------------------------------------------------------------------------

層次遍歷當然用的是廣度優先遍歷了,但是這道題的唯一的一個難點就是怎麼把每一層的元素都寫到一個子容器裡。

要用到兩層迴圈,而且內迴圈的迴圈次數就是每層的元素個數。

class Solution {
public:
    vector<vector<int>> levelOrderBottom(TreeNode* root) {
        vector<vector<int>> s;
        queue<TreeNode*> q;
        if (root==NULL)
            return s;
        //s[0].push_back(root->val);     //這樣寫為什麼不行呢??  不明白。
        q.push(root);
        while(!q.empty())
        {
            int len=q.size();    //計算出每層的元素個數,方便下面的迴圈的次數的統計。


            vector<int>temm;
            for (int j=0;j<len;j++)
           {
            TreeNode * tem=q.front();
            temm.push_back(tem->val);
            if (tem->left!=NULL)
            {
               q.push(tem->left);
            }
            if (tem->right!=NULL)
            {
                q.push(tem->right);
            }
             q.pop();
           }
            s.push_back(temm);
        }
    reverse(s.begin(),s.end());    //將容器的內容進行反置。。。。。

     return s;
    }
};

 

 

這道題的思路還是中的內迴圈真的沒想到,沒想到佇列還有一個size()函式,來記錄它的裡面的元素的個數。