“侧影”就是从左侧或者右侧去观察物体所看到的内容。例如上图中男生的侧影是从他右侧看过去的样子,叫“右视图”;女生的侧影是从她左侧看过去的样子,叫“左视图”。
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
思路:
本题思路就是先按照中序以及后序遍历去建树,然后进行层序遍历。
代码:
C++代码:
#include <iostream>
#include <queue>
using namespace std;
const int N = 25;
int n, cnt, inorder[N], postorder[N], deep;
int res[N][N];
queue <int> q;
struct node {
int value, l = -1, r = -1;
}nodes[N];
int build(int inorder_l, int inorder_r, int postorder_l, int postorder_r) {
if (inorder_l > inorder_r) return -1; // 如果没有节点,则返回 -1
// 获取后序遍历的最后一个节点,也将是根节点
int root = postorder[postorder_r], idx = cnt ++ ;
nodes[idx].value = root;
int pos = inorder_l;
// 获取中序遍历,根节点所在位置
while (inorder[pos] != root) pos++;
// 构建左子树
nodes[idx].l = build(inorder_l, pos - 1, postorder_l, postorder_l + (pos - 1 - inorder_l));
// 构建右子树
nodes[idx].r = build(pos + 1, inorder_r, postorder_l + pos - inorder_l, postorder_r - 1);
return idx;
}
int main() {
// 输入
cin >> n;
for (int i = 1; i <= n; i ++ ) cin >> inorder[i];
for (int i = 1; i <= n; i ++ ) cin >> postorder[i];
// 建树
build(1, n, 1, n);
// 层序遍历
q.push(0);
while (!q.empty()) {
// 根据当前这层的数量来循环
for (int i = q.size(); i > 0; i -- ) {
int j = q.front(); q.pop();
// 每层的结果按照从左到右储存到res里面
res[deep][ ++ res[deep][0]] = nodes[j].value;
if (nodes[j].l != -1) q.push(nodes[j].l);
if (nodes[j].r != -1) q.push(nodes[j].r);
}
// 加深深度
deep ++ ;
}
// 格式化输出
cout << "R:";
for (int i = 0; i < deep; i ++ ) cout << ' ' << res[i][res[i][0]];
cout << endl << "L:";
for (int i = 0; i < deep; i ++ ) cout << ' ' << res[i][1];
cout << endl;
}