原题链接:F - 二叉树的遍历
给出一棵二叉树的中序和前序遍历,输出它的后序遍历。
Input
本题有多组数据,输入处理到文件结束。
每组数据的第一行包括一个整数n,表示这棵二叉树一共有n个节点。
接下来的一行每行包括n个整数,表示这棵树的中序遍历。
接下来的一行每行包括n个整数,表示这棵树的前序遍历。
3<= n <= 100
Output
每组输出包括一行,表示这棵树的后序遍历。
Sample Input
7
4 2 5 1 6 3 7
1 2 4 5 3 6 7
Sample Output
4 5 2 6 7 3 1
前序遍历:当前节点->左节点->右节点
中序遍历:左节点->当前节点->右节点
后序遍历:左节点->右节点->当前节点
首先按前序遍历的顺序依次访问各节点 访问过程中 通过中序遍历 重建以当前节点为根的左子树和右子树
假设前序遍历的当前节点为c c在中序遍历的位置为m m左侧就是c的左子树 右侧就是右子树 然后同理递归
例如当前节点是 [1] 2 4 5 3 6 7 那么当前树的根是1 左右子树是 4 2 5和6 3 7
下一个节点是1 [2] 4 5 3 6 7 那么2的左右子树是4和5 依次类推
每层递归执行复杂度为0n的线性搜索 最坏的情况下复杂度为On^2
#include <iostream>
#include <vector>
#include <string>
#include <algorithm>
using namespace std;
int n,pos;
vector<int>pre,in,post;
void rec(int l,int r)
{
if(l>=r)
return;
int root=pre[pos++];
int m=distance(in.begin(),find(in.begin(),in.end(),root));
rec(l,m);
rec(m+1,r);
post.push_back(root);
}
int main()
{
int k,i;
cin>>n;
for(i=0;i<n;i++)
{
cin>>k;
pre.push_back(k);
}
for(i=0;i<n;i++)
{
cin>>k;
in.push_back(k);
}
pos=0;
rec(0,pre.size());
for(i=0;i<n;i++)
{
if(i)
cout<<" ";
cout<<post[i];
}
cout<<endl;
return 0;
}