L2-006 樹的遍歷 (25分)

L2-006 樹的遍歷 (25分)

AC代碼

#include <iostream>
#include <cstdio>
#include <vector>
#include <algorithm>
using namespace std;
struct node{
    int id, index, level;
};
vector<node> post, in, pre;
void pre_order(int root, int startt, int endd, int index, int level){
    if (startt > endd) return;
    int i = startt;
    while (i < endd && in[i].id != post[root].id) i++;
    pre.push_back({post[root].id, index, level});
    pre_order(root+i-endd-1, startt, i-1, index*2+1, level+1);
    pre_order(root-1, i+1, endd, index*2+2, level+1);
}
bool cmp(node x, node y){
    if (x.level != y.level) return x.level < y.level;
    return x.index < y.index;
}
int main(){
    int n;
    scanf("%d", &n);
    post.resize(n);
    in.resize(n);
    for (int i = 0; i < n; i++)
        scanf("%d", &post[i].id);
    for (int i = 0; i < n; i++)
        scanf("%d", &in[i].id);
    pre_order(n-1, 0, n-1, 0, 0);
    sort(pre.begin(), pre.end(), cmp);
    for (int i = 0; i < n; i++){
        if (i) printf(" ");
        printf("%d", pre[i].id);
    }
    return 0;
}