题目:
“侧影”就是从左侧或者右侧去观察物体所看到的内容。例如上图中男生的侧影是从他右侧看过去的样子,叫“右视图”;女生的侧影是从她左侧看过去的样子,叫“左视图”。
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
感谢用户DSA修正数据!
代码长度限制
16 KB
时间限制
400 ms
内存限制
64 MB
栈限制
8192 KB
思路分析:
已知后序序列和中序序列,由后序序列的性质可以知道,最后一个为根,然后再在中序序列里找根,分出左右子树,以此类推 -> 递归
找区间的时候,先推出较为简单的区间,al ~ ar ,较难的区间用 “知二求一” 可知,在后序序列找的区间长度和在中序序列找的区间长度是一样的(即左右子树的区间是固定的)
代码:
#include<iostream>
#include<cmath>
#include<vector>
#include<algorithm>
using namespace std;
int n,a[105],b[105];
vector<pair<int,int> > res[105];
void build(int al,int ar,int bl,int br,int level,int num)
{
if (al > ar) return;
res[level].push_back({num,b[br]});
int i;
for (i = 1; a[i] != b[br]; ++i) ; // 在中序序列找根
//区间长度确定, 用前面a数组的区间推b数组的区间
build(al,i-1,bl,bl+(i-1-al),level+1,num*2); // 左子树
build(i+1,ar,br-1-(ar-i-1),br-1,level+1,num*2+1); // 右子树
}
int main()
{
cin>>n;
for (int i = 1; i <= n; ++i) cin>>a[i];
for (int i = 1; i <= n; ++i) cin>>b[i];
build(1,n,1,n,1,1);
cout<<"R:";
for (int i = 1; res[i].size() > 0; i++)
{
//pair排序的时候优先以pair的第一个元素为比较值
sort(res[i].begin(), res[i].end()); //对当前层的节点进行排序
cout<<" "<<res[i].back().second;
}
cout<<endl<<"L:";
for (int i = 1; res[i].size() > 0; i++)
{
cout<<" "<<res[i].front().second;
}
return 0;
}