给你一个二叉树,请你返回其按 层序遍历 得到的节点值。 (即逐层地,从左到右访问所有节点)。
示例:
二叉树:[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
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
class Solution {
public:
struct mem{
TreeNode* root;
int layer;
};
vector<vector<int>> levelOrder(TreeNode* root) {
if(root==NULL)
{
return {};
}
queue<mem> q;
mem m_root;
m_root.root=root;
m_root.layer=0;
q.push(m_root);
vector<vector<int>> v;
int vSize=0;
v.resize(vSize);
while(!q.empty())
{
mem curRoot = q.front();
q.pop();
if(curRoot.root->left!=NULL)
{
mem tmpRoot;
tmpRoot.root=curRoot.root->left;
tmpRoot.layer=curRoot.layer+1;
q.push(tmpRoot);
}
if(curRoot.root->right!=NULL)
{
mem tmpRoot;
tmpRoot.root=curRoot.root->right;
tmpRoot.layer=curRoot.layer+1;
q.push(tmpRoot);
}
if(vSize<curRoot.layer+1)
{
vSize++;
v.resize(vSize);
v[curRoot.layer].push_back(curRoot.root->val);
}
else
{
v[curRoot.layer].push_back(curRoot.root->val);
}
}
return v;
}
};