藍橋杯訓練：簡單樹——先序和中序遍歷轉出後序遍歷

原創

2020-02-22 12:16

問題描述：給出二叉樹的先序，中序遍歷序列，求出其後序遍歷序列
作者：何知令
完成時間：2017年8月1日
輸入：首行輸入該二叉樹節點數量，隨後一行輸入該二叉樹前序序列，後一行輸入中序序列
輸出：該二叉樹後序序列
算法思想：找根節點，在左子樹中遞歸，在右子樹中遞歸

代碼：

/*
問題描述：給出二叉樹的先序，中序遍歷序列，求出其後序遍歷序列
作者：何知令
完成時間：2017年8月1日
輸入：首行輸入該二叉樹節點數量，隨後一行輸入該二叉樹前序序列，後一行輸入中序序列
輸出：該二叉樹後序序列
算法思想：找根節點，在左子樹中遞歸，在右子樹中遞歸
*/
#include <iostream>
#include <malloc.h>
using namespace std;
char pre[100],in[100];
typedef struct TreeNode
{
    struct TreeNode *lchild;
    struct TreeNode *rchild;
    char data;
} BTNode;
BTNode * CreateBT(char *pre,char *in,int n)
{
    BTNode *b;
    char r,*p;
    int k;
    if(n<=0)return NULL;
    r=*(pre);//根節點值
    b=(BTNode *)malloc(sizeof(BTNode));//申請一個BTNode類型的存儲空間
    b->data=r;
    for(p=in; p<in+n; p++)
        if(*p==r)
            break;
    k=p-in;//該根節點的左子樹節點數量
    b->lchild=CreateBT(pre+1,in,k);//對其左子樹進行建樹
    b->rchild=CreateBT(pre+k+1,p+1,n-k-1);//對其右子樹進行建樹
    return b;
}
void postVistBT(BTNode *b)
{
    if(b!=NULL)
    {
        preVistBT(b->lchild);
        preVistBT(b->rchild);
        cout<<b->data<<' ';
    }
}
int main()
{
    int n,i;
    BTNode *bt;
    cin>>n;
    for(i=0; i<n; i++)
        cin>>pre[i];
    for(i=0; i<n; i++)
        cin>>in[i];
    bt=CreateBT(pre,in,n);
    postVistBT(bt);
    return 0;
}