L2-006 树的遍历 (25分)

在后序遍历和中序遍历->树的过程中，把层序遍历需要的信息填上即可（层数+出现前后）

需要层数

#include<bits/stdc++.h>
using namespace std;
int post[35],inn[35];
struct node{
    int data;
    int order;int layer;
};
bool cmp(node a,node b){
    if(a.layer!=b.layer) return a.layer<b.layer;
    else return a.order<b.order;
}
vector<node> v;int cnt=0;
void dfs(int lp,int rp,int li,int ri,int length,int layer){
    if(length<=0) return;
    int root=post[rp];
    v.push_back(node{root,cnt++,layer});
    int j=li;
    while(inn[j]!=root) j++;
    int left=j-li;int right=ri-j;
    dfs(lp,lp+left-1,li,j-1,left,layer+1);
    dfs(rp-right,rp-1,j+1,ri,right,layer+1);
}
int main(){
    int n;
    cin>>n;
    for(int i=0;i<n;i++) cin>>post[i];
    for(int i=0;i<n;i++) cin>>inn[i];
    dfs(0,n-1,0,n-1,n,0);
    sort(v.begin(),v.end(),cmp);
    for(int i=0;i<v.size();i++){
        if(i!=0) cout<<" ";
        cout<<v[i].data;
    }
    return 0;
}

不需要层数的是这样，用完全二叉树的规则标记

void pre(int root, int start, int end, int index) {
     if(start > end) return ;
     int i = start;
     while(i < end && in[i] != post[root]) i++;
     level[index] = post[root];
     //这样是不care层数的
     pre(root - 1 - end + i, start, i - 1, 2 * index + 1);
     pre(root - 1, i + 1, end, 2 * index + 2);
}