【PTA数据结构 | C语言版】还原二叉树

本专栏持续输出数据结构题目集，欢迎订阅。

题目

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

输入格式:
输入首先给出正整数 n（≤50），为树中结点总数。随后 2 行先后给出前序和中序遍历序列，均是长度为 n 的不包含重复英文字母（区别大小写）的字符串。

输出格式:
输出为一个整数，即该二叉树的高度。

输入样例:
9
ABDFGHIEC
FDHGIBEAC

输出样例:
5

代码

#include <stdio.h>
#include <stdlib.h>
#include <string.h>

#define MAX_N 50

// 从先序和中序序列递归构建二叉树并计算高度
int buildTreeAndGetHeight(char *preorder, char *inorder, int n) {
    if (n == 0) return 0;
    
    // 先序序列的第一个字符是根节点
    char root = preorder[0];
    
    // 在中序序列中找到根节点的位置
    int rootPos = 0;
    while (inorder[rootPos] != root) rootPos++;
    
    // 递归计算左子树和右子树的高度
    int leftHeight = buildTreeAndGetHeight(preorder + 1, inorder, rootPos);
    int rightHeight = buildTreeAndGetHeight(preorder + 1 + rootPos, inorder + rootPos + 1, n - rootPos - 1);
    
    // 当前树的高度为左右子树高度的最大值加1
    return (leftHeight > rightHeight ? leftHeight : rightHeight) + 1;
}

int main() {
    int n;
    char preorder[MAX_N + 1], inorder[MAX_N + 1];
    
    scanf("%d", &n);
    scanf("%s %s", preorder, inorder);
    
    int height = buildTreeAndGetHeight(preorder, inorder, n);
    printf("%d\n", height);
    
    return 0;
}