二叉树的层次遍历

给你一个二叉树,请你返回其按 层序遍历 得到的节点值。 (即逐层地,从左到右访问所有节点)。

示例:

二叉树:[3,9,20,null,null,15,7],

3
/
9 20
/
15 7
返回其层序遍历结果:

[

[3],

[9,20],

[15,7]

]

/**
* Definition for a binary tree node.
* struct TreeNode {
* int val;
* TreeNode *left;
* TreeNode *right;
* TreeNode() : val(0), left(nullptr), right(nullptr) {}
* TreeNode(int x) : val(x), left(nullptr), right(nullptr) {}
* TreeNode(int x, TreeNode *left, TreeNode *right) : val(x), left(left), right(right) {}
* };
/
class Solution {
public:
vector<vector> levelOrder(TreeNode
root) {

//判断root是否为空
if(root == nullptr) return {};
vector<vector> vec;

//层次遍历,建立一个队列,保证先进先出原则
queue<TreeNode*> q;
q.push(root);
while(!q.empty()){

unsigned size = q.size();
vec.push_back(vector());
//遍历每一层的节点并将下一层的节点放入队列中
for(unsigned i = 0; i < size; ++i){

auto node = q.front();
q.pop();
vec.back().push_back(node->val);
if(node->left)
q.push(node->left);
if(node->right)
q.push(node->right);

}
}
return vec;
}
};

posted @ 2021-10-26 21:58  shared_pointer  阅读(164)  评论(0)    收藏  举报