leetcode中的二叉树级顺序遍历

Question

尝试运行代码“抛出'std :: bad_alloc'what（）：whatstd :: bad_alloc的实例后调用终止”时出错：代码：

#include<bits/stdc++.h>
using namespace std;

class Solution {
public:

    vector<vector<int>> levelOrder(TreeNode* root) {
        int d = height(root);
        vector<int> lol[d];
        vector<vector<int>> ans;
        queue<TreeNode*> q;
        if(root!=NULL)
            q.push(root);
        for(int i=0; i<=d;i++){
            ans.push_back(level(i,q,lol[i]));
        }
        return ans;
    }

    vector<int> level(int i,queue<TreeNode*> q,vector<int> lol){
        if (q.size() != 0)
        {
            for(int a=0;a<=i;a++){
                lol.push_back(q.front()->val);
                q.push(q.front()->left);
                q.push(q.front()->right);
                q.pop();
            }
        }
        return lol;
    }

    int height(TreeNode* root){
        if (root==NULL)
            return 0;
        else{
            int lef = height(root->left) + 1;
            int rig = height(root->right) + 1;
            if(lef>rig)
                return lef;
            else
                return rig;
        }
    }
};

输出应采用以下格式：[[3]，[9,20]，[15,7]]

谁能告诉我代码中的错误在哪里。 我是C ++和leetcode编程的新手，所以很难发现错误。 请帮忙。

Answer 1

我的猜测是，您将节点推入队列而没有检查它们是否为NULL。 从队列弹出时，您也可以取消引用而不检查它们。