队列BFS实现:
简单嘛,创建一个队列一层一层出队入队的就行了。
class Solution {
public:vector<vector<int>>list;
vector<vector<int>> levelOrder(TreeNode* root) {
//秒杀题。。。就是个队列的BFS。。
queue<TreeNode*>q;
if(root!=nullptr)
q.push(root);
while(q.size()){
int size=q.size();
vector<int>t;
while(size){
TreeNode* s = q.front();
q.pop();
t.push_back(s->val);
if(s->left!=nullptr)
q.push(s->left);
if(s->right!=nullptr)
q.push(s->right);
size--;
}
list.push_back(t);
}
return list;
}
};
用DFS伪BFS:
实际上的代码运行原理还是dfs,只是用参数记录层数,第一遍深入用于创建存放每一层的数据结构,后面根据层数把相应的值存入每层对应的数据结构中来。
class Solution {
public:
vector<vector<int>>list;
vector<vector<int>> levelOrder(TreeNode* root) {
dfs(root,0);
return list;
}
private:
//n用于记录此时的遍历处于第几层
void dfs(TreeNode* root,int n){
//触底就不执行
if(root==nullptr)
return;
//由于dfs会一直往深处遍历,所以第一次触底的时候就是构建好二维数组列表个数,后面根据处于第几层将数据分配到二维数组的哪一行
if(list.size()==n){
vector<int>t;
list.push_back(t);
}
list[n].push_back(root->val);
//对左右子树的正常dfs
dfs(root->left,n+1);
dfs(root->right,n+1);
}
};