练习

7-10 树的遍历

 

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

输入格式：

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

输出格式：

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

输入样例：

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

 

输出样例：

4 1 6 3 5 7 2

#include <iostream>
#include <vector>
using namespace std;
vector <int> post,in,level(10000,-1);//申请10000个空间 ，每个值都是-1
//分别设计向量来存储后序遍历，中序遍历，已经层序遍历
int n;
void ergodic(int root,int start,int end,int index)
{
    //遍历 
    //root为根节点，start，end表示子树的范围，index记录树的位置
    int i=start;
    if(start>end)
     return;//终止循环
     while(i<end&&in[i]!=post[root]) 
       i++;//找到根在中序遍历中的位置 
    level[index]=post[root];//把子树的根放在level相应的位置上
    ergodic(root-1-end+i,start,i-1,2*index+1) ;
    ergodic(root-1,i+1,end,2*index+2);//右子树 
}
int main() {
    cin>>n;
    post.resize(n);//为了节约内存，重新设置vector大小 
    in.resize(n);
    for(int i=0;i<n;i++)
    cin>>post[i];
    for(int i=0;i<n;i++)
    cin>>in[i];
    ergodic(n-1,0,n-1,0);
    for(int i=0,cnt=0;i<level.size();i++)
    {
        if(level[i]!=-1)
        {
            cout<<level[i];
            cnt!=n-1?cout<<' ':cout<<endl;
            cnt++;
        }
    }
    return 0;
}