数据结构实验--还原二叉树

一、问题描述

给定一棵二叉树的先序遍历序列和中序遍历序列，要求计算该二叉树的高度。

 

二、输入/输出

输入说明：输入正整数N，为树中结点总数。另外给出先序遍历序列和中序遍历序列。

输出说明：输出一个整数，即该二叉树的高度。

 

三、代码

 

# include <stdio.h>
# include <malloc.h>

typedef struct tagNode
{
    char data;
    struct tagNode *left;//左孩子
    struct tagNode *right;//右孩子
}BitTree;

BitTree * PreIn(char *pre, char *in, int n)
{

    int i = 0;
    BitTree *BT = NULL;
    BitTree *left = NULL;
    BitTree *right = NULL;
    if(n == 0) return NULL;

    BT = (BitTree*)malloc(sizeof(BitTree));
    BT->data = pre[0];

    for(i = 0; i < n; i++)
    {
        if(pre[0] == in[i])
            break;
    }

    left = PreIn(pre+1, in, i);
    right = PreIn(pre+i+1, in+i+1, n-i-1);

    BT->left = left;
    BT->right = right;

    return BT;
}

int GetBitTreeHight(BitTree *BT)
{
    int HL, HR, MaxH;
    if(BT)
    {
        HL = GetBitTreeHight(BT->left);
        HR = GetBitTreeHight(BT->right);

        MaxH = HL > HR?HL : HR;

        return MaxH + 1;
    }

    else
    {
        return 0;
    }
}
int main()
{
    int size;
    int h;
    char pre[50];
    char in[50];
    BitTree * BT;
    printf("请输入结点个数:");
    scanf("%d",&size);
    printf("
请输入先序序列
");
    scanf("%s", pre);
    printf("
请输入中序序列
");
    scanf("%s", in);

    BT = PreIn(pre, in, size);

    h = GetBitTreeHight(BT);
    printf("二叉树高度为%d", h);
    return 0;


}

参考：http://blog.csdn.net/cwqbuptcwqbupt/article/details/6874622