原题:牛客
题目描述
给定一个二叉树,返回该二叉树层序遍历的结果,(从左到右,一层一层地遍历)
例如:
给定的二叉树是{3,9,20,#,#,15,7},
该二叉树层序遍历的结果是
[
[3],
[9,20],
[15,7]
]
示例1
输入
{
1,2}
返回值
[[1],[2]]
示例2
输入
{
1,2,3,4,#,#,5}
返回值
[[1],[2,3],[4,5]]
说明:本题目包含复杂数据结构TreeNode,点此查看相关信息
思想:
本例考量二叉树的层次遍历,首先题目给出了该二叉树的结构,是动态二叉树;
通过本题的输出样例可知,每一层的输出结果单独放在了一个数组当中,且根据题目给出的提示,使用变长数组存放每一层的遍历结果较为合适,并且最后的结果也说明,最后的结果是存放在一个二维数组中的;
因此,创建一个变长的二维数组ans存放最终的结果,用一个一维变长数组res存放每一层的遍历结果;
二叉树的层次遍历,首先要将根节点入队列,由于给出的是动态二叉树,因此队列中存放的是一个结构体,当队列中没有元素的时候,直接返回结果(必须注意这一点!)为了确保当前这一层已经遍历结束,使用size标记当前队列里结构体的元素个数;
队列当中,遵循先入先出的原则,因此,每次入队一个结构体,之后遍历这个结构体的左右子树的结构体,如果不是空则要一直往队列里面加,每次遍历到一个结点,便输出该结点的值;
代码:
/**
* struct TreeNode {
* int val;
* struct TreeNode *left;
* struct TreeNode *right;
* };
*/
#include <queue>
class Solution {
public:
/**
*
* @param root TreeNode类
* @return int整型vector<vector<>>
*/
vector<vector<int> > levelOrder(TreeNode* root) {
// write code here
vector<vector<int> > ans; //存放最终结果
if(!root)
return ans;
queue<TreeNode*> q; //队列存放结构体
q.push(root); //将根节点入队
while(!q.empty()) //当队列不为空时
{
int size = q.size(); //获取当前队列中的内容多少
vector<int> res; //该数组用来存放当前队列的遍历结果;
while(size--)
{
TreeNode * font = q.front(); //获取当前队列的头结点
q.pop(); //该结点出队
res.push_back(font->val); //将该结点的值压入存放当前队列的数组
if(font->left != NULL) q.push(font->left); //递归遍历左子树
if(font->right != NULL) q.push(font->right); //递归遍历右子树
}
ans.push_back(res); //存放最终结果
}
return ans;
}
};