天梯赛L2-3 浪漫侧影 (25 分)

题目描述

“侧影”就是从左侧或者右侧去观察物体所看到的内容。例如上图中男生的侧影是从他右侧看过去的样子，叫“右视图”；女生的侧影是从她左侧看过去的样子，叫“左视图”。

520 这个日子还在打比赛的你，也就抱着一棵二叉树左看看右看看了……

我们将二叉树的“侧影”定义为从一侧能看到的所有结点从上到下形成的序列。例如下图这棵二叉树，其右视图就是 { 1, 2, 3, 4, 5 }，左视图就是 { 1, 6, 7, 8, 5 }。

 

于是让我们首先通过一棵二叉树的中序遍历序列和后序遍历序列构建出一棵树，然后你要输出这棵树的左视图和右视图。

输入格式：

输入第一行给出一个正整数 N (≤20)，为树中的结点个数。随后在两行中先后给出树的中序遍历和后序遍历序列。树中所有键值都不相同，其数值大小无关紧要，都不超过 int 的范围。

输出格式：

第一行输出右视图，第二行输出左视图，格式如样例所示。

输入样例：

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

输出样例：

R: 1 2 3 4 5
L: 1 6 7 8 5

思路：

左视图即从左往右看每一层，第一个看到的结点（实际为每层最左边结点）共同构成了左视图。

右视图即从右往左看每一层，第一个看到的结点（实际为每层最右边结点）共同构成了右视图。

我们可以根据中序和后序遍历序列按层建树，再分别输出该层的首尾结点即可。

AC代码

#include <iostream>
#include <algorithm>
#include <vector>
using namespace std;

const int N = 50;
int in[N], post[N];
int mlevel;
vector<int> v[N];

void sol(int in[], int post[], int length, int level) {
    if (length <= 0) return ;
    int target = post[length - 1];
    int i;
    for (i = 0; i < length; i++) {
        if (in[i] == target) break;
    }

    v[level].push_back(target);
    mlevel = max(mlevel, level);

    sol(in, post, i, level + 1);
    sol(in + i + 1, post + i, length - i - 1, level + 1);
}

int main() {
    int n;
    cin >> n;
    for (int i = 0; i < n; i++) cin >> in[i];
    for (int i = 0; i < n; i++) cin >> post[i];

    sol(in, post, n, 0);

    cout << "R:";
    for (int i = 0; i <= mlevel; i++) {
        cout << ' ' << v[i].back();
    }
    cout << endl;
    cout << "L:";
    for (int i = 0; i <= mlevel; i++) {
        cout << ' ' << v[i].front();
    }
    return 0;
}