B1086 Tree Traversals Again (栈模拟

B1086 Tree Traversals Again (25分)

由题意可以推出，每次访问一个新的结点时就把它入栈，这个过程和先序序列总是先访问根节点的性质是相同的，因此Push的次序就是先序遍历中元素的顺序，Pop则是按左子树，根节点，右子树的顺序进行的，因此Pop的次序是中序遍历序列中元素的顺序.

#include<iostream>
#include<bits/stdc++.h>
#include<stack>
#include<cstring>
#include<string.h>


using namespace std;
const int MAX=1e6+10;

string s;
stack<int>st;
int num=0,n;
int pre[MAX],in[MAX];

struct node
{
    int data;
    node* rchild;
    node* lchild;
};

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->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);
    printf("%d",root->data);
    num++;
    if(num<n)
        printf(" ");
}

int main()
{
    scanf("%d",&n);
    int preindex=0,inindex=0;
    for(int i=0;i<n*2;i++)
    {
        cin>>s;
        if(s=="Push")
        {
            int x;
            scanf("%d",&x);
            st.push(x);
            pre[preindex++]=x;
        }
        else
        {
            in[inindex++]=st.top();
            st.pop();
        }
    }
    node* root=create(0,n-1,0,n-1);
    postorder(root);
    return 0;
}