问题思考:
- 一眼层序遍历,取每层的第一个元素。
- 层序遍历需要有一颗二叉树,又给了中序遍历和前序遍历,需要构造一颗二叉树,套模板即可。但需要额外注意,自己写的时候,pre的边界容易写错。
- 层序遍历用一个队列实现,那么随着队列元素不断弹出,如何分辨此时temp到了下一层呢?
- 给节点加一个level属性,入队的元素肯定在temp节点的下一层。每次入队的时候计算节点的level
- 用一个cur变量记录当前已经取到第几层了,cur从0开始。
代码实现:
#include<iostream>
#include<vector>
#include<queue>
using namespace std;
struct node {// 不熟悉静态写法,乖乖定义数据结构就好了
int data, level;
struct node *lchild, *rchild;
};
vector<int> res;
int n, in[25], pre[25];
node *build(int in_arr[], int in_l, int in_r, int pre_arr[], int pre_l, int pre_r) {
if (in_l > in_r) return nullptr;
int pos = in_l;
while (in_arr[pos] != pre_arr[pre_l] && pos <= in_r) pos++;
node *root = new node;
root->data = in_arr[pos];
root->lchild = build(in_arr, in_l, pos - 1, pre_arr, pre_l + 1, pre_l + pos - in_l);
// root->rchild = build(in_arr, pos + 1, in_r, pre_arr, pos + 1, pre_r); // 错误示范,pre跟着pos走,肯定要走偏
root->rchild = build(in_arr, pos + 1, in_r, pre_arr, pre_l + pos - in_l + 1, pre_r); // 正确姿势,pre有它自己的基准
return root;
}
int main() {
srand(time(0));
scanf("%d", &n);
for (int i = 0; i < n; i++) {
scanf("%d", in + i);
}
for (int i = 0; i < n; i++) {
scanf("%d", pre + i);
}
node *root = build(in, 0, n - 1, pre, 0, n - 1);
queue<node*> que;
root->level = 0;
que.push(root);
res.push_back(root->data);
int cur = 0;
while (que.size()) {
node *temp = que.front();
que.pop();
if (temp->level > cur) {
res.push_back(temp->data);
cur++;
}
if (temp->lchild) {
temp->lchild->level = temp->level + 1;
que.push(temp->lchild);
}
if (temp->rchild) {
temp->rchild->level = temp->level + 1;
que.push(temp->rchild);
}
}
for (int i = 0; i < res.size(); i++) {
i && printf(" ");
printf("%d", res[i]);
}
return 0;
}