L2-006 树的遍历 （25 分）

7
2 3 1 5 7 6 4
1 2 3 4 5 6 7

4 1 6 3 5 7 2

#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
const int maxn = 1e6 + 10;
struct node
{
    int data;
    struct node *lc, *rc;
};
int a[maxn],b[maxn];
struct node *creat(int a[], int b[], int n) // A是中序，B是后序
{
    struct node *root;
    if(n <= 0) return NULL; // 最后的
    int i = 0;
    root = (struct node *)malloc(sizeof(struct node ));
    root -> data = b[n - 1]; //后序的最后一个是根节点
    for(i = 0; i < n; i ++)
    {
        if(a[i] == b[n - 1]) break; // 找到中序的这个点，就是左右子树的分界线
    }
    root -> lc = creat(a,b,i); // 左子树
    root -> rc = creat(a + i + 1,b + i,n - i - 1); // 右子树，A中序来说是根节点右边一个开始，中序来说就是右边这些，长度要减去根和右边的
    return  root;
}
void level(struct node *root)
{
    if(root != NULL)
    {
        queue<node*>q;
        q.push(root);
        bool f = 1;
        while(!q.empty())
        {
            struct node *x;
            x = q.front();
            q.pop();
            if(f)printf("%d", x -> data),f = 0;
            else printf(" %d", x->data);
            if(x->lc)q.push(x->lc);
            if(x->rc)q.push(x->rc);
        }
    }
    printf("\n");
}
int main()
{
    int n;
    scanf("%d", &n);
    for(int i = 0; i < n; i ++) scanf("%d", &a[i]);
    for(int i = 0; i < n; i ++) scanf("%d", &b[i]);
    struct node *root;
    root = creat(b,a,n);
    level(root);
    return 0;
}