leetcode筆記：Binary Tree Level Order Traversal

leetcode筆記：Binary Tree Level Order Traversal

一. 題目描述

Given a binary tree, return the level order traversal of its nodes’ values. (ie, from left to right, level by level).
For example: Given binary tree {3,9,20,#,#,15,7},

  3
 / 
 9 20
   / 
  15  7

二. 題目分析

無。

三. 示例代碼

參考了Discussion中stellari的做法，遞歸進行層次遍歷，並將每個level對應於相應的vector：

#include 
#include 

using namespace std;

struct TreeNode
{
    int val;
    TreeNode *left;
    TreeNode *right;
    TreeNode(int x) :val(x), left(NULL), right(NULL){}
};

class Solution 
{
public:
    vector > levelOrder(TreeNode *root)
    {
        vector > result;
        orderTraversal(root, 1, result);
        return result;
    }

private:
    void orderTraversal(TreeNode* root, int level, vector > & result)
    {
        if (root == NULL) return;

        if (level > result.size()) // 每往下一級，創建一個空的vector保存該層數據
            result.push_back(vector());

        result[level - 1].push_back(root->val);

        orderTraversal(root->left, level + 1, result);
        orderTraversal(root->right, level + 1, result);
    }
};

四. 小結

以上為遞歸實現，也可以寫成迭代版本，時間復雜度O(n)，空間復雜度O(1)。