L2-3 浪漫侧影 传统做法 带注释

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

分析：

这是经典的二叉树题，按照常规流程做即可，这里我使用传统的链式结构存储二叉树。

首先根据中序后序建树，建好之后进行dfs，获取每一层节点的分布，并存入二位数组g中。最后对g进行解析，取每一行的头尾元素即为左视图和右视图元素。

代码：

 

#include <bits/stdc++.h>
using namespace std;
struct node
{
	int data;
	node* lchild;
	node* rchild;
};

int mid[25];		//存放中序 
int pos[25];		//存放后序 
int maxlevel=0;		//二叉树层数 
vector <int> g[25];	//记录每层的节点排布 
int N;				//节点数 

node* create(int m1,int m2,int p1,int p2)	//四个值分别对应中序数组和后序数组的首尾下标	
{
	//递归出口 
	if(p1>p2)	return NULL;
	//新建节点 
	node* root =new node;
	root->data=pos[p2];
	//计算偏移量temp 
	int i;
	for(i=m1;i<=m2;i++)
		if(pos[p2]==mid[i])
			break;
	int temp=i-m1;
	//递归建树 
	root->lchild=create(m1,i-1,p1,p1+temp-1);
	root->rchild=create(i+1,m2,p1+temp,p2-1);
	//返回建好的树 
	return root;
}

//dfs，获取maxlevel和g 
void dfs(node* r,int now)	//now代表当前层数 
{
	if(r==NULL)	return ;
	g[now].push_back(r->data);
	if(now>maxlevel)
		maxlevel=now;
	dfs(r->lchild,now+1);
	dfs(r->rchild,now+1);
} 


int main()
{
	cin>>N;
	for(int i=0;i<N;i++)
		cin>>mid[i];
	for(int i=0;i<N;i++)
		cin>>pos[i];
		
	node* root = create(0,N-1,0,N-1);
	
	dfs(root,1);			
	
	vector<int> v1,v2;		//v1v2分别存放左视图和右视图列表 
	for(int i=1;i<=maxlevel;i++)
	{
		v1.push_back(g[i][0]);	//每行第一个为左视图元素 
		v2.push_back(g[i][g[i].size()-1]);	//最后一个为右视图元素 
	}
	
	//按规则输出 
	cout<<"R:";
	for(int i=0;i<v2.size();i++)
		cout<<" "<<v2[i]; 
	puts("");
	cout<<"L:";
	for(int i=0;i<v1.size();i++)
		cout<<" "<<v1[i]; 
	
}