二叉树重建与镜像层序遍历的实现与优化

今忱

1. 题目解析与核心思路

这道题目考察的是二叉树的基本操作和变形处理。给定一棵二叉树的中序遍历和前序遍历序列，要求我们首先重建这棵二叉树，然后输出其镜像后的层序遍历序列。这实际上包含了三个关键操作：根据遍历序列重建二叉树、对二叉树进行镜像反转、进行层序遍历输出。

1.1 二叉树重建原理

二叉树的重建是基于前序遍历和中序遍历的特性完成的。前序遍历的第一个元素总是当前子树的根节点，而中序遍历中根节点的位置可以将序列分为左子树和右子树两部分。通过递归地应用这一原理，我们可以完整地重建整棵二叉树。

在实际操作中，为了提高效率，我们使用哈希表（unordered_map）来存储中序遍历中每个节点的位置，这样可以在O(1)时间内找到根节点在中序遍历中的位置，避免了每次线性查找的时间消耗。

1.2 镜像反转的实现技巧

题目要求的"反转"或"镜像"操作，实际上就是交换每个节点的左右子树。在代码实现上，我们不需要真正修改树的结构，只需要在层序遍历时改变入队顺序即可。常规的层序遍历是先左后右，而镜像后的层序遍历则是先右后左。

这种实现方式非常巧妙，它避免了显式地进行树结构的修改，而是通过遍历顺序的改变达到了同样的效果，既节省了时间又减少了空间的使用。

2. 代码实现详解

2.1 数据结构定义与初始化

cpp复制#include<bits/stdc++.h>
#define lc p<<1       // 左孩子
#define rc (p<<1)+1   // 右孩子
using namespace std;

// 模拟二叉树存储结构
int tr[100010];
int a[5000], b[5000];
int idx = 1;
unordered_map<int, int> mp;

这里我们使用了数组来模拟二叉树结构，这是一种常见的空间优化方法。通过位运算（p<<1）来快速计算左右孩子的索引位置，这在处理完全二叉树时特别高效。哈希表mp用于存储中序遍历中各个节点的位置信息，以便快速查找。

2.2 二叉树重建函数

cpp复制void f(int l, int r, int p) {
    if(l > r) {
        return;
    }
    // 前序遍历的当前元素为根节点
    tr[p] = b[idx++];
    
    // 获取根节点在中序序列中的位置
    int md = mp[tr[p]];
    
    // 递归构建左子树和右子树
    f(l, md - 1, lc);
    f(md + 1, r, rc);
}

这个递归函数是重建二叉树的核心。参数l和r定义了当前子树在中序遍历序列中的范围，p是当前节点在数组tr中的索引。每次递归调用都会处理一个子树，直到范围无效（l > r）时返回。

2.3 镜像层序遍历实现

cpp复制queue<int> q;
q.push(1);
bool fl = 0; // 控制输出格式前的空格

while(q.size()){
    int p = q.front();
    q.pop();
    
    // 如果当前节点为空，则跳过
    if(tr[p] == 0) continue; 
    
    if(fl) cout << " ";
    cout << tr[p];
    fl = 1;
    
    // 核心：因为要反转，所以先压入右孩子，再压入左孩子
    q.push(rc);
    q.push(lc);
}