【PTA】Tree Traversals Again (25 分)

最新推荐文章于 2021-07-29 15:40:19 发布

原创最新推荐文章于 2021-07-29 15:40:19 发布 · 210 阅读

1 ·

CC 4.0 BY-SA版权

PTA_数据结构专栏收录该内容

33 篇文章

订阅专栏

本文介绍了一种使用栈实现的非递归方法，通过预序和中序遍历构建二叉搜索树。通过`CreateT`函数，根据输入序列创建节点，并利用`postOrder`进行后序遍历输出。算法适用于处理动态输入的数据结构问题。

#include <iostream>
#include <stack>
#include <string.h>
#define MAXN 30
using namespace std;

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


int in[MAXN],pre[MAXN];
int in_index=0,pre_index=0;

node* CreateT(int preL,int preR,int inL,int inR){
    if(preL>preR) return NULL;

    int tmp=pre[preL]; 
    node* root=new node;
    root->data=tmp;
       
    int j;
    for(j=inL;j<=inR;j++){
        if(in[j]==tmp) break;
    }

    root->lchild=CreateT(preL+1,preL+j-inL,inL,j-1);
    root->rchild=CreateT(preL+j-inL+1,preR,j+1,inR);

    return root;
}

int p=0;
void postOrder(node* root){
    if(!root) return;
	postOrder(root->lchild);
	postOrder(root->rchild);
	if(p)printf(" %d",root->data);
	else {
		printf("%d",root->data);
		p=1;
	}
}


//二叉树的非递归遍历
int main(){
    int N;
    scanf("%d\n",&N);

    string str;int op;
    stack<int> s;
    for(int i=0;i<2*N;i++){
        cin>>str;
        if(str=="Push"){
//只有push才需要二次读取数字，不能指定str[位数]，可能是两位数以上，二次读取最易
            cin>>op;
            s.push(op);
//push()时输出是先序
            pre[in_index++]=op;
        }else{
            op=s.top();
            s.pop();
//pop()时输出时中序
            in[pre_index++]=op;
        }
    }


    node* root=CreateT(0,N-1,0,N-1);

    postOrder(root);


    return 0;
}