还原二叉树

OJ传送门

/*
 两个字符串，其长度n均小于等于26。
 第一行为前序遍历，第二行为中序遍历。
 求出其后序遍历
 */
#include <stdio.h>
#include <string.h>

static struct Node{
    Node *lchild;
    Node *rchild;
    char c;
}Tree[110];
static int loc;
static char str1[30],str2[30];

static Node *creat(){
    Tree[loc].lchild=Tree[loc].rchild=NULL;
    return &Tree[loc++];
}

static void postOrder(Node *T){
    if (T->lchild!=NULL)
        postOrder(T->lchild);
    if (T->rchild!=NULL)
        postOrder(T->rchild);
    printf("%c",T->c);
}

Node *build(int s1,int e1,int s2,int e2){
    Node* ret =creat();
    ret->c=str1[s1];
    int rootIdx;
    for (int i=s2;i<=e2;i++) {
        if (str2[i]==str1[s1]) {
            rootIdx=i;
            break;
        }
    }
    if (rootIdx!=s2) {
        ret->lchild=build(s1+1, s1+rootIdx-s2, s2, rootIdx-1);
    }
    if (rootIdx!=e2) {
        ret->rchild=build(s1+rootIdx-s2+1, e1, rootIdx+1, e2);
    }
    return ret;
}

int main(){
    while (scanf("%s",str1)!=EOF) {
        scanf("%s",str2);
        loc=0;
        int L1=(int)strlen(str1);
        int L2=(int)strlen(str2);
        Node *T=build(0, L1-1, 0, L2-1);
        postOrder(T);
        printf("\n");
    }
    return 0;
}