给你一个二叉树,请你返回其按 层序遍历 得到的节点值。 (即逐层地,从左到右访问所有节点)。
示例:
二叉树:[3,9,20,null,null,15,7],3
/ \
9 20
/ \
15 7
返回其层序遍历结果:[
[3],
[9,20],
[15,7]
]来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/binary-tree-level-order-traversal
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
实现思路:
利用队列进行bfs
实现代码:
class Solution {
public:
vector<vector<int>> levelOrder(TreeNode* root) {
vector<vector<int>> ret;
if(root==NULL) return ret;
queue<TreeNode*> q;
q.push(root);
while(!q.empty()){
int cnt=q.size();
ret.push_back(vector<int>());
for(int i=0;i<cnt;i++){
auto node=q.front();
q.pop();
ret.back().push_back(node->val);
if(node->left) q.push(node->left);
if(node->right) q.push(node->right);
}
}
return ret;
}
};
这道题我收获了这样几点,首先是学习容器嵌套时如何处理,如vector<vector<int>>,对于这样的容器,首先要把一个空的容器push_back()进去,然后向其back()中继续push_back()元素,也就是向新加入的容器中加入元素。
同时要注意q.push()先要判断是否为空。