![在这里插入图片描述](https://i-blog.csdnimg.cn/blog_migrate/3529762dfac81dc6e7c88f20b9e285c6.png)
题目大意
给出二叉树的后序和中序遍历序列,让你输出层序遍历结果。
思路解析
从后序序列中找父节点,然后在中序序列中找到该父节点的左右两棵子树,每下一层要记录该层的层数,供输出使用,递归执行这个过程。这样遍历的结果是先序序列,节点的趋势是从上向下、从左向右的,所以将某层的节点按序输出就是该层的层序序列。所以递归算法是核心。过程如下:因为后序的最后一个总是根结点,令i在中序中找到该根结点,则i把中序分为两部分,左边是左子树,右边是右子树。左子树在后序中的根结点为root – (end – i + 1),即为当前根结点-右子树的个数。左子树在中序中的起始点为start,末尾end点为i – 1.右子树的根结点为当前根结点的前一个结点root – 1,右子树的起始点start为i+1,末尾end点为end。
示例代码
#include<iostream>
#include<vector>
using namespace std;
int n, maxdeep = 0;
vector<int> post(n), in(n);
vector<vector<int>> level(30,vector<int>());
void creat(int root, int start,int end,int deep) {
if (start > end) {
return;
}
if (deep > maxdeep) maxdeep = deep;
level[deep].push_back(post[root]);
int i = start;
for (i; i < end && in[i] != post[root]; i++);
creat(root-(end-i+1), start, i-1,deep+1);
creat(root - 1, i + 1, end,deep+1);
}
int main() {
scanf("%d", &n);
for (int i = 0; i < n; i++) {
int temp;
scanf("%d", &temp);
post.push_back(temp);
}
for (int i = 0; i < n; i++) {
int temp;
scanf("%d", &temp);
in.push_back(temp);
}
creat(n-1,0,n-1,0);
for (int i = 0; i <= maxdeep; i++) {
for (int j = 0; j < level[i].size(); j++) {
if (i == maxdeep && j == level[i].size() - 1) {
continue;
}
printf("%d ", level[i][j]);
}
}
printf("%d", level[maxdeep][level[maxdeep].size() - 1]);
return 0;
}