2021-04-23

数据结构（树）

题目地址：http://acm.ticknet.hnust.cn/status.php?user_id=2005060214&cid=2276

题目大意：给出两个序列，一个为树的前序遍历，一个为树的中序遍历，要求跟据前序遍历和中序遍历构建一棵树，然后通过后序遍历输出这颗树。

思路：前序遍历：先访问根节点，然后前序遍历左子树，再前序遍历右子树。
中序遍历：中序遍历根节点的左子树，然后是访问根节点，最后中序遍历右子树。

因为二叉树的前序遍历序列一定是该树的根节点，而中序遍历序列中根节点前面一定是该树的左子树，后面是该树的右子树
所以可以先通过前序遍历找到根节点，并构建根节点，然后在中序遍历中找到这个根节点，并以该根节点的位置将中序遍历分成两半，左边部分的是左子树，右边部分是右子树，假设左子树有x个元素，右子树有y个元素，那么继续从前序遍历的前x个元素中按顺序找左子树的根，当然前面以及找过的不用再继续找了，从前序遍历的后y个元素中按顺序找右子树的根，同样已经找过的不用继续找了，最后不断进行上述操作，即不断递归即可

代码：
开始是这样写的，但不符合要求（我把题目给的函数接口改变了）
题目给的函数接口：

#include<iostream>
#include<cstdlib>
#include<cstdio>
#include<cstring>
using namespace std;
 
typedef char TElemType;
//二叉树的二叉链表存储表示
typedef struct BiTNode
{
    TElemType data;                      //结点数据域
    struct BiTNode *lchild, *rchild; //左右孩子指针
} BiTNode, *BiTree;
 
//根据先序序列pre[pre_low..pre_low+len-1]和中序序列in[in_low..in_low+len-1]建树t
void BuildTree(BiTree& t, char pre[], int pre_low, char in[], int in_low, int len,int pp)
{
    /****在此下面完成代码************/
    int pos;
    if(pre_low>=pp)
    {
        t=NULL;
        return;
    }
    t=new BiTNode;
    t->data=pre[pre_low];
    for(int i=in_low; i<len; i++)
    {
        if(in[i]==t->data)
        {
            pos=i;
            break;
        }
    }
    BuildTree(t->lchild,pre,pre_low+1,in,in_low,pos,pre_low+pos-in_low+1);
    BuildTree(t->rchild,pre,pre_low+pos+1-in_low,in,pos+1,len,pp);
    /*********************************/
}
// 后序遍历的递归算法
void PostOrderTraverse(BiTree t)
{
    /****在此下面完成代码************/
    if(t)
    {
        PostOrderTraverse(t->lchild);
        PostOrderTraverse(t->rchild);
        cout<<t->data;
    }
    /*********************************/
}
 
void DestroyBitree(BiTree& t)
{
    /****在此下面完成代码************/
    if(t)
    {
        DestroyBitree(t->lchild);
        DestroyBitree(t->rchild);
        delete t;
    }
    /*********************************/
}
 
int main()
{
    char pre[30], in[30];
    BiTree t = NULL;
    while(cin >> pre)
    {
        cin >> in;
        BuildTree(t, pre, 0, in, 0, strlen(in),strlen(in));
        PostOrderTraverse(t);
        DestroyBitree(t);
        cout << endl;
    }
}
 

后面想了一下，构建树的函数中pp完全没有必要pp=len+pre_low-in_low

#include<iostream>
#include<cstdlib>
#include<cstdio>
#include<cstring>
using namespace std;
 
typedef char TElemType;
//二叉树的二叉链表存储表示
typedef struct BiTNode
{
    TElemType data;                      //结点数据域
    struct BiTNode *lchild, *rchild; //左右孩子指针
} BiTNode, *BiTree;
 
//根据先序序列pre[pre_low..pre_low+len-1]和中序序列in[in_low..in_low+len-1]建树t
void BuildTree(BiTree& t, char pre[], int pre_low, char in[], int in_low, int len)
{
    /****在此下面完成代码************/
    int pos;
    if(pre_low>=len+pre_low-in_low)
    {
        t=NULL;
        return;
    }
    t=new BiTNode;
    t->data=pre[pre_low];
    for(int i=in_low; i<len; i++)
    {
        if(in[i]==t->data)
        {
            pos=i;
            break;
        }
    }
    BuildTree(t->lchild,pre,pre_low+1,in,in_low,pos);
    BuildTree(t->rchild,pre,pre_low+pos+1-in_low,in,pos+1,len);
    /*********************************/
}
// 后序遍历的递归算法
void PostOrderTraverse(BiTree t)
{
    /****在此下面完成代码************/
    if(t)
    {
        PostOrderTraverse(t->lchild);
        PostOrderTraverse(t->rchild);
        cout<<t->data;
    }
    /*********************************/
}
 
void DestroyBitree(BiTree& t)
{
    /****在此下面完成代码************/
    if(t)
    {
        DestroyBitree(t->lchild);
        DestroyBitree(t->rchild);
        delete t;
    }
    /*********************************/
}
 
int main()
{
    char pre[30], in[30];
    BiTree t = NULL;
    while(cin >> pre)
    {
        cin >> in;
        BuildTree(t, pre, 0, in, 0, strlen(in));
        PostOrderTraverse(t);
        DestroyBitree(t);
        cout << endl;
    }
}

也可以改变递归结束条件

#include<iostream>
#include<cstdlib>
#include<cstdio>
#include<cstring>
using namespace std;

typedef char TElemType;
//二叉树的二叉链表存储表示
typedef struct BiTNode
{
    TElemType data;                      //结点数据域
    struct BiTNode *lchild, *rchild; //左右孩子指针
} BiTNode, *BiTree;

//根据先序序列pre[pre_low..pre_low+len-1]和中序序列in[in_low..in_low+len-1]建树t
void BuildTree(BiTree& t, char pre[], int pre_low, char in[], int in_low, int len)
{
    if (len<=0)
    {
        t=NULL;
        return;
    }
    t=new BiTNode;
    t->data=pre[pre_low];
    int pos=0;//这里一定要赋初值为0
    while(in[in_low+pos]!=t->data)
        pos++;
    BuildTree(t->lchild,pre,pre_low+1,in,in_low,pos);
    BuildTree(t->rchild,pre,pre_low+pos+1,in,in_low+pos+1,len-(pos+1));
    return;
}
void PostOrderTraverse(BiTree t)
{
    /****在此下面完成代码************/
    if(t)
    {
        PostOrderTraverse(t->lchild);
        PostOrderTraverse(t->rchild);
        cout<<t->data;
    }
    /*********************************/
}

void DestroyBitree(BiTree& t)
{
    /****在此下面完成代码************/
    if(t)
    {
        DestroyBitree(t->lchild);
        DestroyBitree(t->rchild);
        delete t;
    }
    /*********************************/
}

int main()
{
    char pre[30], in[30];
    BiTree t = NULL;
    while(cin >> pre)
    {
        cin >> in;
        BuildTree(t, pre, 0, in, 0, strlen(in));
        PostOrderTraverse(t);
        DestroyBitree(t);
        cout << endl;
    }
}