L2-2. 树的遍历
时间限制
400 ms
内存限制
65536 kB
代码长度限制
8000 B
判题程序
Standard
作者
陈越
给定一棵二叉树的后序遍历和中序遍历,请你输出其层序遍历的序列。这里假设键值都是互不相等的正整数。
输入格式:
输入第一行给出一个正整数N(<=30),是二叉树中结点的个数。第二行给出其后序遍历序列。第三行给出其中序遍历序列。数字间以空格分隔。
输出格式:
在一行中输出该树的层序遍历的序列。数字间以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 <string>
#include <sstream>
#include <queue>
#define MAX 35
using namespace std;
int In[MAX],Out[MAX],ltr[MAX],rtr[MAX];
int n;
int build_tr(int l1,int r1,int l2,int r2)
{
if(l1 > r1) return 0;
int root = Out[r2];
int cur=l1;
while(In[cur] != root)
cur++;
int cnt = cur-l1;
ltr[root] = build_tr(l1,cur-1,l2,l2+cnt-1); //???????????
rtr[root] = build_tr(cur+1,r1,l2+cnt,r2-1); //???????????
return root;
}
void Out_put()
{
queue<int > a;
int temp;
int root=Out[n-1];
int flag=0;
a.push(root);
while(!a.empty()){
temp=a.front();
if(!flag){
cout<<temp;
flag=1;
}
else
cout<<" "<<temp;
a.pop();
if(ltr[temp]) a.push(ltr[temp]);
if(rtr[temp]) a.push(rtr[temp]);
}
cout<<endl;
}
int main()
{
while(cin>>n)
{
for(int i = 0;i < n;i ++) cin>>Out[i];
for(int i = 0;i < n;i ++) cin>>In[i];
build_tr(0,n-1,0,n-1);
Out_put();
}
return 0;
}