跟32的区别是,每一层的要单独进一个数组。所以需要一个flag位置来判断是否这层已经结束了。
flag位置的加法:上一层最后一个点进去的时候,加完左右子树加入flag
1
2 3
45 67
先读root,1进队之后输入null,2,3进队。
1出队遇到null,于是2,3后加入null;
2出队,45进队;3出队遇到null,于是67进队,加入null
32-3类似,在指定层的时候 reverse一下就可以。
class Solution {
public:
vector<vector<int>> levelOrder(TreeNode* root) {
vector<vector<int>> result={};
if (root==NULL)
return result;
queue<TreeNode*>que;
que.push(root);
que.push(NULL);
vector<int >res={};
while(!que.empty()){
TreeNode* x=que.front();
que.pop();
if(x==NULL){
if(res.size()){
result.push_back(res);
res={};
que.push(NULL);
}else break;
continue;
}
res.push_back(x->val);
if(x->left)
que.push(x->left);
if(x->right)
que.push(x->right);
}
return result;
}
};
网友评论