“侧影”就是从左侧或者右侧去观察物体所看到的内容。例如上图中男生的侧影是从他右侧看过去的样子,叫“右视图”;女生的侧影是从她左侧看过去的样子,叫“左视图”。
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
思路:
- 中、后序转层序
- 输出每一层的第一个数和最后一个数
#include<iostream>
#include<cstdio>
using namespace std;
int post[1048577],in[1048577],level[1048577];
int n;
void dfs(int start,int end,int root,int index){
if(start > end)
return;
int i = start;
while(i < end && in[i]!=post[root])
i ++;
//i代表的是中序遍历中根节点的位置
level[index] = post[root];
dfs(start,i-1,root-(end-i)-1,index*2+1);//左子树
dfs(i+1,end,root-1,index*2+2);//右子树
}
int main(){
cin >> n;
for(int i = 0 ; i < n ; i ++)
cin >> in[i];
for(int i = 0 ; i < n ; i ++)
cin >> post[i];
dfs(0,n-1,n-1,0);
cout << "R: " << level[0];
long long i = 1 , j = 2;
for(int t = 2 ; t <= 20 ; t ++){
//从每一层的结束的结点位置开始输出
for(long long k = j ; k >= i ; k --)
{
if(level[k])
{
printf(" %d",level[k]);
break;
}
}
int temp = 1 << t;//第t层的最大结点数
i += temp/2;//最开始的结点位置
j += temp;//结束的结点位置
}
printf("\nL: %d",level[0]);
i = 1 , j = 2;
for(int t = 2 ; t <= 20 ; t ++){
//从每一层的最开始的结点开始输出
for(long long k = i ; k <= j ; k ++)
{
if(level[k]){
printf(" %d",level[k]);
break;
}
}
int temp = 1 << t;
i += temp/2;
j += temp;
}
}