看到L2级别的题关于考察二叉树的遍历的题目还不少,好像是有3个吧。
这个题,给你树的后续和中序遍历,输出层次遍历。
没头没绪!T_T
就手抄代码,一个类似模板的东西。
使用vector存储后序遍历、中序遍历和层次遍历的数值,post[]、in[]和level[]。
主要部分是递归函数f(root,start,end,index):
index是遍历顺序树的结点的索引,用来存放到level中,树的两个子节点的索引分别是2index+1和2index+2。
递归函数f从start开始,到end结束,在后序顺序post中找到对应的结点,放入level中,两个递归分别让start = i+1和end = i-1缩小范围。
其中root决定了level要取的结点在post中的位置。
总之
f(root - 1 - end + i, start, i - 1, 2 * index + 1)
f(root - 1, i + 1, end, 2 * index + 2)
前序或者后序可以确定root的位置,从定义开始推,递归思想确定两个字树。
最后,附代码:
#include<iostream>
#include<vector>
using namespace std;
vector<int> post,in,level(100000,-1);
int N;
void f(int root,int start,int end,int index)
{
int i = start;
if(start > end)
return;
while(i<end&&in[i]!=post[root])
i++;
level[index] = post[root];
f(root-1-end+i,start,i-1,2*index+1);
f(root-1,i+1,end,2*index+2);
}
int main()
{
cin >> N;
post.resize(N);
in.resize(N);
for(int i=0;i<N;i++)
cin >> post[i];
for(int i=0;i<N;i++)
cin >> in[i];
f(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;
}