7-4 树的遍历 (20 分) 后序+中序恢复层序

参考链接:👉https://blog.csdn.net/shiliang97/article/details/102924283
带点注释理解一下

题目

思路

层序按行输出，对树的分析类似于完全二叉树，不好分层就利用完全二叉树的下标
具体分析见注释

#include <bits/stdc++.h>
using namespace std;
vector<int> post,in,level(100000,-1 ); //要在多个函数间调用，所以声明为全局比较好
int n;

void ergodic(int root,int start,int end,int index)
{//利用后序来找根root，
    int i=start;
    if(start > end) return;
    while(i<end && in[i]!=post[root]) i++; //在中序中找根的下标
    level[index] = post[root]; //提前计算好index的值，将当前节点放到lecel中
    ergodic(root-1-end+i,start,i-1,2*index+1); //左子树，新root同之前恢复计算公式，index是从0开始的完全树下标计算公式
    ergodic(root-1,i+1,end,2*index+2);  //右子树
}

int main()
{
    cin >> n;
    post.resize(n);
    in.resize(n);
    for(int i=0;i<n;i++)
        cin >> post[i];
    for(int i=0;i<n;i++)
        cin >> in[i];
    ergodic(n-1,0,n-1,0);
    for(int i=0,cnt=0;i<level.size();i++)
    {
        if(level[i]!=-1) //-1说明未赋值
        {
            cout << level[i];
            cout << ((cnt!=n-1)? " ":"\n");  //i用于遍历，cnt记录已经记录了几个
            cnt++;
        }
    }
}