1. 程式人生 > >Leetcode:二叉樹的層序遍歷、前中後序遍歷非遞迴版

Leetcode:二叉樹的層序遍歷、前中後序遍歷非遞迴版

102. 二叉樹的層次遍歷

給定一個二叉樹,返回其按層次遍歷的節點值。 (即逐層地,從左到右訪問所有節點)。

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

    3
   / \
  9  20
    /  \
   15   7

返回其層次遍歷結果:

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

典型的層序遍歷,套模板:

    vector<vector<int>> levelOrder(TreeNode* root)
    {
        vector<vector<int>>result;
        if (root == nullptr)return result;
        queue<TreeNode*>que;
        que.push(root);
        while (!que.empty())
        {
            int size = que.size();//保證佇列裡始終是同一層的資料
            vector<int> temp;
            for (int i = 0; i < size; i++)
            {
                TreeNode* node = que.front();
                que.pop();                
                temp.push_back(node->val);
                if (node->left != NULL)que.push(node->left);
                if (node->right != NULL)que.push(node->right);
            }
            result.push_back(temp);
        }
        return result;
    }