1. 非递归层序遍历
算法思想:
1. 先将根节点入队,此时队列不为空,开始循环;
2. 计算出队列的长度size,为了确保for循环一轮循环size次,也就是树的每一层的元素个数
3. 让node指针永远指向队头元素,为了暂存队头结点并将其记录保存下来,以便于执行出队操作后,依然能对出队后的结点进行操作,也就是让node指针的左右结点依次按顺序入队。
4. 再将左孩子结点和右孩子结点入队。进行下一次循环时,由于根结点已经出队,则此时que.size() = 2,for循环执行两次。此时result = [[1],[2,3]];队列结点元素分别为:【4,5,6,7】
5. 下一次循环后,que.size() 已经被重新赋值为4。现在队列里的结点都是叶子节点,所以两个if条件不成立;循环完成后,vec = [4,5,6,7],最后再把该次的vec追加到result容器中,则直接返回result = [[1],[2,3],[4,5,6,7]]; 层序遍历结束。
代码实现如下:
void levelOrder(TreeNode* root){
vector<vector<int>> result;
queue<TreeNode*> que;
if(root != nullptr) que.push(root); //先把根节点入队
while(!que.empty()){
int size = que.size(); //长度不是固定的,会随着队列元素个数变化
vector<int> vec; //子容器
for(int i = 0;i < size;i++){
TreeNode* node = que.front(); //先把对头元素保存下来
que.pop(); //出队访问
vec.push_back(node->data);
if(node->lchild) que.push(node->lchild); //分别让该节点的孩子入队
if(node->rchild) que.push(node->rchild);
}
result.push_back(vec); //每结束一层,把每一层放入结果容器中,抽象成一个二维数组
}
for(const auto &internal:result){
for_each(internal.begin(),internal.end(),[](int num){
cout << num << endl;
});
cout << endl;
}
}
2. 层序遍历应用 (翻转二叉树)
注意:
1.层序遍历不光可以应用于翻转二叉树,这里只举出其中一个例子。
2.翻转二叉树也可以使用前序和后序遍历,但是唯独中序遍历不方便
算法思想:
物理上的交换
1. 逻辑和层序遍历相同,代码也大致相同
2. 不同的是:在两个子结点入队之前,需要将二者进行交换,这样就实现了物理上的交换
逻辑上的交换
1. 若我们只是想在遍历的时候,是翻转后的顺序,我们可以不进行交换左右子结点,直接先让右孩子入队,左孩子后入队即可实现逻辑上的交换
代码实现如下:
//使用层序遍历翻转二叉树
static void reverseTree(TreeNode* root){
vector<int> result; //结果容器
queue<TreeNode*> que; //临时保存节点队列
if(root != nullptr) que.push(root);
while(!que.empty()){
int size = que.size();
for(int i = 0;i < size;i++){
TreeNode* node = que.front(); //先保存对头元素
que.pop();//对头元素出队
swap(node->lchild,node->rchild); //让左孩子和右孩子调换
if(node->lchild) que.push(node->lchild);
if(node->rchild) que.push(node->rchild);
}
}
for_each(result.begin(),result.end(),[](int num){
cout << num << endl;
});
}