PTA数据结构与算法题目集(中文) 7-23

PTA数据结构与算法题目集(中文)  7-23

7-23 还原二叉树 (25 分)

 

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

输入格式:

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

输出格式:

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

输入样例:

9
ABDFGHIEC
FDHGIBEAC

输出样例:

5
题目分析：利用前序遍历与中序遍历 计算树的高度 利用前序遍历找树节点 然后在中序遍历中限制范围
这个我写的最后一个测试点无法通过

#define _CRT_SECURE_NO_WARNINGS
#include<stdio.h>

int Find(char c, char str[], int Begin,int End)
{
    for (int i = Begin; i < End; i++)
    {
        if (c == str[i])
            return i;
    }
    return -1;
}

int GetHeight(char str1[], char str2[], int Length)
{
    int Height = 0;
    int Begin = 0;
    int Flag = 0;
    int End = Length;
    int i = Find(str1[Flag], str2, Begin, End);
    if (i == -1)
        return Height;
    Height++;
    while (1)
    {
        if (End - Begin <= 0)
            return Height;
        if (i - Begin >=End - i-1)
        {
            End = i;
            Flag++;
        }
        else
        {
            Flag += i - Begin + 1;
            Begin = i + 1;
        }
        i = Find(str1[Flag], str2, Begin, End);
        if (i == -1)
            return Height;
        Height++;
    }
}

int main()
{
    char str1[52];
    char str2[52];
    int N;
    scanf("%d", &N);
    scanf("%s%s", str1,str2);
    printf("%d", GetHeight(str1, str2, N));
    return 0;
}

看了别人写的

#define _CRT_SECURE_NO_WARNINGS
#include<stdio.h>

int DFS(char str1[], char str2[], int n)
{
    if (n == 0)
        return 0;
    int i = 0;
    for (i = 0; i < n && str1[0] != str2[i]; i++)
        ;
    int x = DFS(str1 + 1, str2, i)+1;  //左子树
    int y = DFS(str1 + i + 1, str2 + i+1, n - i-1)+1; //右子树
    return x > y ? x : y;
}
int main()
{
    char str1[101];
    char str2[101];
    int N;
    scanf("%d", &N);
    scanf("%s%s", str1,str2);
    printf("%d", DFS(str1, str2, N));
    return 0;
}