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

参考博客链接

 

递归思想：

先序遍中的X节点，当它出现在中序遍历中时，以中序遍历中X为分界点，它的左边部分为X的左子树，右边部分为X的右子树，然后依次规则递归处理它的左右子树。

核心代码：

用两个数组分别存储前序、中序遍历结果。s1为前序遍历数组中要处理的树的下标起点，e1为终点。同理s2为中序遍历数组中待处理的树的起点，e2为终点。

 

 

Tptr build(int s1, int e1, int s2, int e2) 
{
    Tptr Head = Create();
    Head->data = r1[s1];
    for(int i = s2; i<=e2; i++)
    {　　
        if(i!=s2)Head->left = build(s1+1, s1+i-s2, s2, i-1);//通过计算左右子树长度可以推出待处理树的下标范围，然后作为形参传入
        if(i!=e2)Head->right = build(s1+i-s2+1,e1,i+1,e2);
        break;
    }
    return Head;
}

 

完整AC代码：

#include <iostream>
#include <string>
#include <cstring>
using namespace std;

typedef struct BTNode* Tptr;
char r1[100], r2[100];

struct BTNode
{
    char data;
    Tptr left, right;
};
Tptr Create()
{
    Tptr T = new struct BTNode;
    T->left = T->right = NULL;
    return T;
}
Tptr build(int s1, int e1, int s2, int e2)
{
    Tptr Head = Create();
    Head->data = r1[s1];
    for (int i = s2; i <= e2; i++)
    {
        if (r2[i] == r1[s1])
        {
            if (i != s2)Head->left = build(s1 + 1, s1 + i - s2, s2, i - 1);
            if (i != e2)Head->right = build(s1 + i - s2 + 1, e1, i + 1, e2);     
            break;
        }
    }
    return Head;
}
int GetH(Tptr T)
{
    if (!T)return 0;
    if (!T->left)GetH(T->left);
    if (!T->right)GetH(T->right);
    return GetH(T->left) > GetH(T->right) ? GetH(T->left) + 1 : GetH(T->right) + 1;
}
int main()
{
    int n;
    cin >> n;
    cin >> r1;
    cin >> r2;
    Tptr T = build(0, n - 1, 0, n - 1);
    cout << GetH(T);
    return 0;
}