LeetCode 102

https://leetcode.cn/problems/binary-tree-level-order-traversal/description/

难度:中等

这道题是 BFS 的经典题目。本题使用两个数组 cur 和 nxt 来实现 BFS。

时间复杂度:O(n),其中 n 为二叉树的节点个数。

空间复杂度:O(n)。满二叉树(每一层都填满)最后一层有大约 n/2 个节点,因此数组中最多有 O(n) 个元素,所以空间复杂度是 O(n) 的。

class Solution {
public:
    vector<vector<int>> levelOrder(TreeNode* root) {
        if (root == nullptr) {
            return {};
        }
        vector<TreeNode*> cur;
        cur.push_back(root);
        vector<vector<int>> ans;
        while (cur.size() > 0) {
            vector<TreeNode*> nxt;
            vector<int> level;
            for (TreeNode* node: cur) {
                level.push_back(node->val);
                if (node->left) {
                    nxt.push_back(node->left);
                }
                if (node->right) {
                    nxt.push_back(node->right);
                }
            }
            ans.push_back(level);
            cur = move(nxt);
        }
        return ans;
    }
};