问题描述:给定二叉树的后序和中序,给出层次遍历。
解题思路:根据后序最后一个为根,在中序中找出左右子树,递归建树。然后层次遍历输出。
方法二:建树的时候加上高度,并将顶点加入该高度的集合。然后直接输出即可。
AC代码:
#include<bits/stdc++.h>
using namespace std;
vector<vector<int> >level(30);//每一层的顶点集合
vector<int>post(30),in(30);
int N;
void tolevel(int inL,int inR,int root,int depth)
{
if(inL>inR) return;
int i=inL;
level[depth].push_back(post[root]);
while(i<inR&&in[i]!=post[root])i++;//在中序中找根
tolevel(inL,i-1,root-(inR-i)-1,depth+1);//左子树,高度加一,
tolevel(i+1,inR,root-1,depth+1);//右子树,
}
int main()
{
int i,j,count=0;
freopen("test.txt","r",stdin);
scanf("%d",&N);
for(i=0;i<N;++i)scanf("%d",&post[i]);
for(i=0;i<N;++i)scanf("%d",&in[i]);
tolevel(0,N-1,N-1,0);
i=0;
while(level[i].size()){//层次输出
for(j=0;j<level[i].size();++j){
printf("%d",level[i][j]);
if(count<N-1)printf(" ");
++count;
}
++i;
}
return 0;
}