题目大意:
用栈的输出表示二叉树的中序遍历,要求输出二叉树的后序遍历序列。
解题思路:
题目隐含了入栈的顺序就是先序遍历的序列,根据这个构建出二叉树然后后序输出即可。
代码如下:
#include<iostream>
#include<cstdio>
#include<fstream>
#include<set>
#include<cmath>
#include<cstring>
#include<string>
#include<map>
#include<vector>
#include<iomanip>
#include<cstdlib>
#include<list>
#include<queue>
#include<stack>
#include<algorithm>
#define inf 0x3f3f3f3f
#define MOD 1000000007
#define mem0(a) memset(a,0,sizeof(a))
#define mem1(a) memset(a,-1,sizeof(a))
#define meminf(a) memset(a,inf,sizeof(a))
//vector ::iterator it;
//set<int>::iterator iter;
typedef long long ll;
typedef unsigned long long ull;
using namespace std;
int pre[50],in[50],n,num=1;
struct node
{
int data;
node* lchild;
node* rchild;
};
node* create(int prel,int prer,int inl,int inr)//由前序,中序构建二叉树
{
if(prel>prer)return NULL;
node *root=new node;
root->data=pre[prel];
int k;
for(int i=inl;i<=inr;i++)
{
if(in[i]==pre[prel])
{
k=i;
break;
}
}
int numleft=k-inl;
root->lchild=create(prel+1,prel+numleft,inl,k-1);
root->rchild=create(prel+numleft+1,prer,k+1,inr);
return root;
}
void postorder(node* root)//后序遍历
{
if(root==NULL)return;
//左右根
postorder(root->lchild);
postorder(root->rchild);
cout<<root->data;
num++;
if(num<=n)cout<<' ';
}
int main()
{
// std::ios::sync_with_stdio(false);
// cin.tie(0);
// freopen("test.txt","r",stdin);
// freopen("output.txt","w",stdout);
cin>>n;
stack<int> s;
int indexpre=0,indexin=0;
for(int i=0;i<n*2;i++)
{
string t;
int k;
cin>>t;
if(t=="Push")
{
cin>>k;
pre[indexpre++]=k;//先序序列
s.push(k);
}
else
{
in[indexin++]=s.top();//中序序列
s.pop();
}
}
node* root=create(0,n-1,0,n-1);
postorder(root);
//cout<<endl;
return 0;
}