欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页

PATA 1086 Tree Traversals Again

程序员文章站 2022-07-08 18:33:49
...

PATA 1086 Tree Traversals Again
PATA 1086 Tree Traversals Again

题记

思路:
1.题目中的push顺序其实就是先序遍历的顺序,pop的时候顺便访问,这样得到的顺序其实就是中序遍历顺序,这两者可以唯一确定一棵二叉树
2.对于输入可以一行一行地输入,也可以先输入一个string再输入一个整数,这里我用的是一行一行输入,如果开头是Push那么就把数组前面删掉,然后计算一下数字是多少,压入栈中,顺便把它加到先序遍历数组中;如果是pop就把栈顶元素加到中序遍历数组中并把栈顶元素出栈。
3.之后是常规的后序遍历,递归就可以了,注意末尾不要多加一个空格
注意:
1.题目一定要看清,不要想当然。题目中说的是各个结点的标号是1,2,3…N,但并不是意味着先序遍历就是这个顺序,还得看Push的顺序,如果这一点想错了,就会报段错误,栈会溢出
2.小心递归构建二叉树的时候,左右子树边界传进去的参数加1减1,一定要自己推一遍


代码如下

#include <iostream>
#include <stack>
#include <string>
using namespace std;

const int Maxn=30+5;

int pre[Maxn],in[Maxn],post[Maxn];//前序、中序、后序遍历数组

struct Node{
    int data;
    Node *lchild,*rchild;
};

//输入
int n;

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]==root->data)
            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;
}

int count=0;//防止多输出一个空格
void post_traverse(Node* root){
    if(root==NULL)
        return;
    post_traverse(root->lchild);
    post_traverse(root->rchild);
    count++;
    printf("%d",root->data);
    if(count<n)
        printf(" ");
}

int main()
{
    scanf("%d",&n);
    cin.get();//吃掉换行符
    stack<int> st;

    int p=0,q=0;//记录当前中序遍历到第j个
    for(int i=0;i<2*n;i++){
        string str;//保存读入的每一行
        getline(cin,str);
        if(str.find("Push")!=string::npos){//入栈的操作
            int sum=0;//记录当前入栈的数的值
            str.erase(str.begin(),str.begin()+5);//把"Push "删掉
            while(str.size()!=0){
                sum=sum*10;
                sum+=str[0]-'0';
                str.erase(str.begin());
            }
            pre[p++]=sum;
            st.push(sum);
        }
        else{
            in[q++]=st.top();
            st.pop();
        }
    }

    Node* root=create(0,n-1,0,n-1);
    post_traverse(root);
    printf("\n");
    return 0;
}