CCCC练习 5-10树的遍历

给定一棵二叉树的后序遍历和中序遍历，请你输出其层序遍历的序列。这里假设键值都是互不相等的正整数。

输入格式：

输入第一行给出一个正整数$N$（$\le 30$），是二叉树中结点的个数。第二行给出其后序遍历序列。第三行给出其中序遍历序列。数字间以空格分隔。

输出格式：

在一行中输出该树的层序遍历的序列。数字间以1个空格分隔，行首尾不得有多余空格。

输入样例：

7
2 3 1 5 7 6 4
1 2 3 4 5 6 7

输出样例：

4 1 6 3 5 7 2

 

• 时间限制：400ms
• 内存限制：64MB
• 代码长度限制：16kB
• 判题程序：系统默认
• 作者：陈越
• 单位：浙江大学

#include <stdio.h>
#include <iostream>
#include <vector>
using namespace std;


vector<int> post,in,level(10000,-1);


//根由中序序列中计算得到,start是前序序列的查找开始，end是前序序列的查找末尾
void pre(int root,int start,int end,int index){
int i;
if(start > end) return;
i = start;
while(in[i]!=post[root] && i<end)
i++;
level[index] = post[root];
pre(root-1-end+i,start,i-1,index*2+1);//左子树
pre(root-1,i+1,end,2*index+2);//右子树
}


int main(){
int n;
scanf("%d",&n);
post.resize(n);
in.resize(n);
for(int i=0;i<n;i++)
scanf("%d",&post[i]);
for(int i=0;i<n;i++)
scanf("%d",&in[i]);
pre(n-1,0,n-1,0);
int cnt=0;
for(int i=0;i<level.size();i++){
if(level[i]!=-1 && cnt!=n-1){
printf("%d ",level[i]);
cnt++;
}
else if(level[i]!=-1){
printf("%d\n",level[i]);
break;
}
}
}