题目描述
给你二叉树的根节点 root ,返回其节点值的 层序遍历 。 (即逐层地,从左到右访问所有节点)。
示例 1:
输入:root = [3,9,20,null,null,15,7]
输出:[[3],[9,20],[15,7]]
示例 2:
输入:root = [1]
输出:[[1]]
示例 3:
输入:root = []
输出:[]
算法思想
解法一:
定义一个类型为vector< vector < int > > 类型的返回结果集ans,运用递归思想依次进行visit访问,当层数depth大于等于ans.size()时,说明这一层还没来过,这是第一次来,所以得进行扩容,也就是加括号,然后依次遍历左子树和右子树。
解法二:
利用典型的BFS思想进行代码实现,利用队列实现。
代码实现
解法一代码:
class Solution {
public:
vector<vector<int>> levelOrder(TreeNode* root) {
vector<vector<int>> ans;
visit(root,0,ans);
return ans;
}
void visit(TreeNode *root, int depth, vector<vector<int>> &ans){
if(!root) return;
//depth代表层数,这一层还没来过,这是第一次来,所以要加括号
if(depth >= ans.size())
ans.push_back(vector<int>{});
ans[depth].push_back(root->val);
visit(root->left, depth + 1, ans);
visit(root->right, depth + 1, ans);
}
};
解法二代码:
class Solution {
public:
vector<vector<int>> levelOrder(TreeNode* root) {
vector<vector<int>> res;
if(!root) return res;
queue<TreeNode*> qu;
qu.push(root);
while(!qu.empty())
{
vector<int> tmp;
int len=qu.size();
for(int i=0;i<len;i++){
TreeNode* node=qu.front();
qu.pop();
tmp.push_back(node->val);
if(node->left) qu.push(node->left);
if(node->right) qu.push(node->right);
}
res.push_back(tmp);
}
return res;
}
};