【1086】Tree Traversals Again (25 分)

#include<iostream>
#include<stdio.h>
#include<stdlib.h>
#include<math.h>
#include<string.h>
#include<algorithm>  
#include<map>
#include<vector>
#include<queue>
#include<stack>
using namespace std;  
const int maxn=50;
struct node{
    int data;
    node* lchild;
    node* rchild;
};
int pre[maxn],in[maxn],post[maxn]; //先、中、后序
int n;  //结点个数


//当前二叉树的先序序列区间为[preL,preR],中序序列区间为[inL,inR]
//create函数返回构建出的二叉树的根结点地址
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;//这个k不要在for里面定义了，注意有效域
    for(k=inL;k<=inR;k++){
        if(in[k]==pre[preL]){  //在中序序列中找到in[k]==pre[L]的结点
            break;
        }
    }
    int numLeft=k-inL; //左子树的结点个数
    //返回左子树的根结点地址，赋值给root的左指针
    root->lchild=create(preL+1,preL+numLeft,inL,k-1);
    //返回右子树的根结点地址，赋值给root的右指针
    root->rchild=create(preL+numLeft+1,preR,k+1,inR);
    return root;  //返回根结点地址
}

int num=0;//已输出的结点个数
void postorder(node* root){//后序遍历
    if(root==NULL){
        return;
    }
    postorder(root->lchild);
    postorder(root->rchild);
    printf("%d",root->data);
    num++;
    if(num<n)  printf(" ");
}
int main(){   
    scanf("%d",&n);
    char str[5];
    stack<int> st;
    int x,preIndex=0,inIndex=0;  //入栈元素、先序序列位置及中序序列位置
    for(int i=0;i<2*n;i++){ //出栈入栈共2n次
        scanf("%s",str);
        if(strcmp(str,"Push") == 0){  //入栈
            scanf("%d",&x);
            pre[preIndex++]=x;   //令pre[preIndex]=x
            st.push(x);
        }else{
            in[inIndex++]=st.top();  //令in[inIndex]=st.top
            st.pop();
        }
    }
    node* root=create(0,n-1,0,n-1);  //建树
    postorder(root);  //后序遍历
    system("pause");
    return 0;   
}