题目链接
解法一:
BFS模板题,借助于队列
AC代码:
class Solution {
public:
vector<vector<int>> levelOrder(TreeNode* root) {
// bfs
queue<TreeNode*> que;
vector<vector<int>> res;
if(root!=nullptr) que.push(root);
while(!que.empty()){
int size = que.size();
vector<int> level;
for(int i=0;i<size;i++){
TreeNode* cur = que.front();
que.pop();
level.push_back(cur->val);
if(cur->left!=nullptr) que.push(cur->left);
if(cur->right!=nullptr) que.push(cur->right);
}
res.push_back(level);
}
return res;
}
};
解法二:codetop中提到的面试情况:使用递归实现
DFS,记录所处的层级情况
AC代码:
class Solution {
public:
void travelsal(TreeNode* cur, vector<vector<int>>& res, int level){
if(cur == nullptr) return;
if(level >= res.size()){//说明需要新添加一层
res.push_back(vector<int> {});
}
res[level].push_back(cur->val);
travelsal(cur->left, res, level+1);
travelsal(cur->right, res, level+1);
}
vector<vector<int>> levelOrder(TreeNode* root) {
// dfs
vector<vector<int>> res;
//cout<<res.size()<<endl; //0
travelsal(root, res, 0);
return res;
}
};