Leetcode 102. Binary Tree Level Order Traversal

題目描述

題目鏈接：https://leetcode-cn.com/problems/binary-tree-level-order-traversal/
給定一個二叉樹，返回其按層次遍歷的節點值。 （即逐層地，從左到右訪問所有節點）。
例如:
給定二叉樹: [3,9,20,null,null,15,7],

   3
  / \
 9  20
   /  \
  15   7

返回其層次遍歷結果：

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

解題方案

• 迭代
  設置一個隊列，對二叉樹進行層次遍歷，遍歷到的節點壓入隊列；彈出某個節點後將其左右孩子節點依次壓入隊列中。
  設置兩個變量，記爲last，nlast，其中last用於記錄當前行最右節點，nlast記錄下一行最右節點（nlast始終記錄最新進入隊列的節點）。當last等於最新打印的節點，則是last=nlast，這樣就完成了換行操作。

class Solution {
public:
    vector< vector<int> > levelOrder(TreeNode* root) {
        vector<vector<int>> res;
        if(!root)
            return res;
        vector<TreeNode *> level;
        level.push_back(root);
        while(!level.empty())
        {
            vector<TreeNode *> temp;
            vector<int> level_val;
            for(int i=0;i<level.size();++i)
            {
                if(level[i]->left)
                    temp.push_back(level[i]->left);
                if(level[i]->right)
                    temp.push_back(level[i]->right);
                level_val.push_back(level[i]->val);
            }
            res.push_back(level_val);
            level = temp;
        
        }
        return res;
    }
};

執行用時16ms，內存消耗14MB。

• 遞歸

class Solution {
public:
    vector<vector<int>> levelOrder(TreeNode* root) {
        vector<vector<int>> res;
        preorder(root, 0, res);
        return res;
    }
    void preorder(TreeNode *root, int depth, vector<vector<int>> &res)
    {
        if(!root)
            return;
        if(depth>=res.size())
            res.push_back(vector<int> {});
        res[depth].push_back(root->val);
        preorder(root->left, depth+1, res);
        preorder(root->right, depth+1, res);
            }
        };

執行用時16ms，內存消耗14.7MB。

知識點

• Vector< vector< int> >v 表示二維向量。
• C++ vector 容器淺析