【树】【PAT甲级】1020 Tree Traversals 树的遍历

1020 树的遍历

题目说明：
假设二叉树的所有值都是正数，给出后序和中序遍历序列，要求输出相应的层次遍历序列。（假设所有的正整数都是互不相等的）

输入说明：
每个输入文件包括一个测试用例。在每个例子，第一行会给出总节点数N，总数不超过三十。
第二行是后序序列，第三行是中序序列，所有数字间都用一个空格间隔。

**思路一言以蔽之：**通过后序去找根节点，然后通过中序来确定左右子树。

#include <cstdio>
#include <vector>
#include <map>
using namespace std;

vector<int> post, in;
map<int, int> level;

//中序后序转层次函数
//start和end值是用于确定范围找根的；Index是索引值 
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];
    pre(root - 1 - end + i, start, i - 1, 2 * index + 1);
    pre(root - 1, i + 1, end, 2 * index + 2);
}


int main() {
	
	//用户输入结点数量N 
    int n;
    scanf("%d", &n);
    
    //调节post和In这两个容器向量的大小 
    post.resize(n);
    in.resize(n);
    
    //用户输入树的后序和中序序列 
    for(int i = 0; i < n; i++) scanf("%d", &post[i]);
    for(int i = 0; i < n; i++) scanf("%d", &in[i]);
    
    //调用序列函数 
    pre(n-1, 0, n-1, 0);
    
    auto it = level.begin();
    printf("%d", it->second);
    while(++it != level.end()) printf(" %d", it->second);
    return 0;
}

知识储备

1.STL库

①vector：vector向量容器使用说明

②vector的resize函数：vector.resize(n,element)

相关题目
①给出后序和中序，输出层次遍历序列【本文】；
②给出后序和中序，输出先序序列：柳神后序中序转先序