要求是将层序遍历的结果,第一层中节点的值域存在二维数组的第一个vector中,第二层的存在第二个中。依次继续。
创建一个二维数组作为最终输出结果,创建一个队列用来取每层的拿到的值域,创建一个vector对象levelval用来保存每层拿到的值域的集合。以示例为例,第一次3所在的节点入队列,此时队列中节点的个数size更新为1个,从队列中拿出size个节点,将它的值域交给levelval,如果这个节点有左右孩子,则依次将左右孩子入队列,从队列中删除3这个节点,将levelval插进二维数组ret中。接下来循环直到队列为空,返回ret。
class Solution {
public:
vector<vector<int>> levelOrder(TreeNode* root) {
//创建一个二维数组
vector<vector<int>> ret;
//如果是空树
if(nullptr == root){
return ret;
}
//不是空树,创建队列
queue<TreeNode*> q;
q.push(root);
//队列非空,说明里面有节点
while(!q.empty()){
size_t size = q.size();//队列中节点的个数
vector<int>levelval(size);
for(size_t i = 0; i < size; i++){
TreeNode* cur = q.front();//从队列中拿一个节点
levelval[i] = cur->val;
if(cur->left){
q.push(cur->left);
}
if(cur->right){
q.push(cur->right);
}
q.pop();
}
ret.push_back(levelval);
}
return ret;
}
};