凡是树,首先要考虑能不能用递归的方法来做;比如:
层序:1 2 3 4 5 6 7
中序:4 2 5 1 6 3 7
1必定是根节点,在中序中查找位置,中序中的1将中序分为左右子树,我们用v1存储左子树的节点,然后层序中从第二个节点开始按照层序的顺序依次进入v2,v3;v2,v3分别是左子树和右子树的层序遍历,这样就变成子问题了,
第一次递归这里v2为 2 4 5 ;v3为3 6 7
然后就不用多说了吧,前序和后序操作思想是一样的,大部分代码都一样,就是push_back顺序不一样而已啦
#include <iostream>
#include<stdlib.h>
#include<string>
#include<vector>
#include<set>
#include<math.h>
#include<map>
#include<unordered_map>
#include<algorithm>
#include<functional>
#include<queue>
//头文件有点多,因为不想每次写头文件。。。
using namespace std;
int sum = 0;
vector<int>preorder, inorder, postorder, levelorder;
void getpreorder(vector<int>v,int instart,int inend) {
if (v.size()==0)
return;
int i = 0;
while (inorder[instart + i] != v[0])
i++;
vector<int>v1, v2,v3;
for (int j = instart; j <= instart+i-1; j++) {
v1.push_back(inorder[j]);
}
for (int j = 1; j < v.size(); j++) {
if (find(v1.begin(), v1.end(), v[j]) != v1.end())
v2.push_back(v[j]);
else
v3.push_back(v[j]);
}
preorder.push_back(v[0]);
getpreorder(v2, instart, instart + i - 1);
getpreorder(v3, instart + i + 1, inend);
}
void getpostorder(vector<int>v, int instart, int inend) {
if (instart > inend)
return;
int i = 0;
while (inorder[instart + i] != v[0])
i++;
vector<int>v1, v2, v3;
for (int j = instart; j <= instart + i - 1; j++) {
v1.push_back(inorder[j]);
}
for (int j = 1; j < v.size(); j++) {
if (find(v1.begin(), v1.end(), v[j]) != v1.end())
v2.push_back(v[j]);
else
v3.push_back(v[j]);
}
getpostorder(v2, instart, instart + i - 1);
getpostorder(v3, instart + i + 1, inend);
postorder.push_back(v[0]);
}
int main() {
int n;
scanf("%d", &n);
for (int i = 0; i < n; i++) {
int tem;
scanf("%d", &tem);
levelorder.push_back(tem);
}
for (int i = 0; i < n; i++) {
int tem;
scanf("%d", &tem);
inorder.push_back(tem);
}
getpreorder(levelorder, 0, n - 1);
getpostorder(levelorder, 0, n - 1);
for (int i = 0; i < preorder.size(); i++) {
printf("%d ", preorder[i]);
}
cout << endl;
for (int i = 0; i < postorder.size(); i++) {
printf("%d ", postorder[i]);
}
system("pause");
return 0;
}