push输入的顺序是先序
pop输出的顺序是中序
最后的结果应该是后序输出
两个子函数:
1)先序+中序重建二叉树
2)后序输出二叉树
//pat a 1086
#include<cstdio>
#include<algorithm>
#include<cstring>
#include<stack>
using namespace std;
const int maxn=100;
int pre[maxn],in[maxn],post[maxn];
stack<int> st;
int n;
struct Node{
int data;
Node *left,*right;
};
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(k=inL;k<=inR;k++){
if(in[k]==pre[preL]){
break;
}
}
int numLeft=k-inL;
root->left=create(preL,preL+numLeft-1,inL,k-1);
root->right=create(preL+numLeft+1,preR,k+1,inR);
return root;
}
void postorder(Node *root){
int num=0;
if(root==NULL){
return;
}
postorder(root->left);
postorder(root->right);
printf("%d",root->data);
if(num<n) printf(" ");
}
int main(){
int cnt1=0,cnt2=0;
char s[5];
scanf("%d",&n);
for(int i=0;i<2*n;i++){
scanf("%s",s);
if(strcmp(s,"Push")==0){
scanf("%d",&pre[cnt1++]);
st.push(pre[cnt1++]);
}else{
in[cnt2++]=st.top();
st.pop();
}
}
Node *root=create(0,n-1,0,n-1);
postorder(root);
return 0;
}
坑点:段错误。
复查:
create构建二叉树那边的左指针参数有问题
修改之后无段错误,变成答案错误了(抓狂)
再复查:
修改了一下入栈时的代码,不直接输入pre[cnt1++]里面了,而是使用x用以替代。
AC代码:
//pat a 1086
#include<cstdio>
#include<algorithm>
#include<cstring>
#include<stack>
using namespace std;
const int maxn=100;
int pre[maxn],in[maxn],post[maxn];
stack<int> st;
int n;
struct Node{
int data;
Node *left,*right;
};
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(k=inL;k<=inR;k++){
if(in[k]==pre[preL]){
break;
}
}
int numLeft=k-inL;
root->left=create(preL+1,preL+numLeft,inL,k-1);
root->right=create(preL+numLeft+1,preR,k+1,inR);
return root;
}
int num=0; //已经输出的结点的个数
void postorder(Node *root){ //后序遍历
if(root==NULL){
return;
}
postorder(root->left);
postorder(root->right);
printf("%d",root->data);
num++;
if(num<n) printf(" ");
}
int main(){
int cnt1=0,cnt2=0,x;
char s[5];
scanf("%d",&n);
for(int i=0;i<2*n;i++){
scanf("%s",s);
if(strcmp(s,"Push")==0){
scanf("%d",&x);
pre[cnt1++]=x;
st.push(x);
}else{
in[cnt2++]=st.top();
st.pop();
}
}
Node *root=create(0,n-1,0,n-1);
postorder(root);
return 0;
}