“侧影”就是从左侧或者右侧去观察物体所看到的内容。例如上图中男生的侧影是从他右侧看过去的样子,叫“右视图”;女生的侧影是从她左侧看过去的样子,叫“左视图”。
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
基本思路:
本道题是经典的二叉树遍历问题。首先根据后序遍历和中序遍历创建一个数组,随后开一个多维的vector数组。从二叉树的根结点开始dfs遍历。因为dfs遍历的时候,是从左往右的,所以同一深度的结点在vector[depth]里面一定是按照从左往右的顺序排列的。我们再看题目,不难发现从右边看二叉树实际上就是每个vector[deep]中最右边的结点,即vector[deep]中的最后一个元素;同理从左边看二叉树,每层看到的结点实际上是vector[deep]中的第一个元素。最后按照输出样例输出即可。
参考代码:
#include <iostream>
#include <vector>
using namespace std;
int n, hou[25], zhong[25];
struct tree{
int data;
tree *left,*right;
};
tree* creat(int root, int begin, int end){
if(end < begin)
return NULL;
tree *t = new tree();
t->data = hou[root];
int i;
for(i = begin; hou[root] != zhong[i]; i++);
t->left = creat(root-1-(end-i), begin, i-1);
t->right = creat(root-1, i+1, end);
return t;
}
vector<int> v[105];
void dfs(tree* t, int level)
{
if(t == NULL)
return;
level++;
v[level].push_back(t -> data);
if(t -> left)
dfs(t -> left, level);
if (t -> right)
dfs(t -> right, level);
}
int main(){
scanf("%d", &n);
for(int i = 0; i < n; i++)
scanf("%d", &zhong[i]);
for(int i = 0; i < n; i++)
scanf("%d", &hou[i]);
tree *t = creat(n-1, 0, n-1);
dfs(t, 0);
printf("R:");
for(int i = 1; i <= 20; i++)
{
if(v[i].size() == 0)
break;
printf(" %d", v[i][v[i].size()-1]);
}
printf("\nL:");
for(int i = 1; i <= 20; i++)
{
if(v[i].size() == 0)
break;
printf(" %d", v[i][0]);
}
printf("\n");
return 0;
}