二叉树的层序遍历
顾名思义,二叉树的层序遍历就是从左到右一层一层的遍历二叉树,与先序、中序和后序遍历有所不同。
层序遍历是实现需要借助队列。将二叉树的根节点放入队列,然后出队,访问出队结点并判断它是否有左子树和右子树,先入左子树再入右子树;继续出队,出队时访问出队结点是否有左子树…重复步骤直到队列为空,此时整棵树的结点都被遍历完成。
当本层的左子树结点开始出队时,右子树结点也已经再队列中,再入左子树结点的子结点时不会打乱层序的顺序。
代码实现:
//层序遍历
void LevelOrder(BiTree T)
{
queue<BiTree> q; //借助队列
if (T != NULL)
{
BiTree p; //暂存要出队的结点
q.push(T); //根结点入队 A
while (!q.empty()) //队列非空
{
p = q.front();
q.pop(); //出队
printf("%d ", p->data);
if (p->lchild != NULL) // B
q.push(p->lchild); //队列先进先出,先入左孩子
if (p->rchild != NULL) // C
q.push(p->rchild);
}
}
}
leetcode 102:二叉树的层序遍历
力扣的层序遍历要求每一次的结点存储在一个数组中,因此需要使用size存储当前层数下的结点数
输入:root = [3,9,20,null,null,15,7]
输出:[[3],[9,20],[15,7]]
push_back:
函数将一个新的元素加到vector的最后面,位置为当前最后一个元素的下一个元素
/**
* 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<int>> levelOrder(TreeNode* root) {
queue<TreeNode*> q;
if(root != nullptr) q.push(root);
vector<vector<int>> result; //结果存储
while(!q.empty()){
int size = q.size(); //根据题目要求,记录本层的元素数量
vector<int> vec; //存储本层节点的数组
for(int i = 0; i < size;i++){
TreeNode* p = q.front(); //取队列的第一个元素
q.pop();
vec.push_back(p->val);
if(p->left != nullptr) q.push(p->left);
if(p->right != nullptr) q.push(p->right);
}
result.push_back(vec);
}
return result;
}
};