题目描述:给定一个二叉树,返回该二叉树层序遍历的结果,(从左到右,一层一层地遍历)
题目示例:
该二叉树层序遍历的结果是 [ [3], [9,20], [15,7] ]
解题思路:层序遍历可以利用队列实现。首先将根节点加入队列,依次弹出队首,并将队首节点的左右子节点依次加队尾。依次循环,直至队列为空,即完成对二叉树的层序遍历。注意要设置一变量来判断遍历的层数。C++代码实现如下:
vector<vector<int> > levelOrder(TreeNode* root) {
// write code here
vector<vector<int>> ans;
if(!root)
return ans;
vector<int> an;
queue<TreeNode *>sTree;
sTree.push(root);
while(!sTree.empty())
{
int sTreeSize = sTree.size();
while(sTreeSize>0)
{
TreeNode *node = sTree.front();
sTree.pop();
sTreeSize--;
an.push_back(node->val);
if(node->left!=nullptr)
sTree.push(node->left);
if(node->right!=nullptr)
sTree.push(node->right);
}
ans.push_back(an);
an.clear();
}
return ans;
}