L2-011 玩转二叉树 (25分)

Posted 2022-12-14 绀香零八

L2-011 玩转二叉树 (25分)

给定一棵二叉树的中序遍历和前序遍历，请你先将树做个镜面反转，再输出反转后的层序遍历的序列。所谓镜面反转，是指将所有非叶结点的左右孩子对换。这里假设键值都是互不相等的正整数。

输入格式：
输入第一行给出一个正整数N（≤30），是二叉树中结点的个数。第二行给出其中序遍历序列。第三行给出其前序遍历序列。数字间以空格分隔。

输出格式：
在一行中输出该树反转后的层序遍历的序列。数字间以1个空格分隔，行首尾不得有多余空格。

输入样例：

7
1 2 3 4 5 6 7
4 1 3 2 6 5 7

输出样例：

4 6 1 7 5 3 2

Code：
无须构造二叉树进行广度优先搜索
·使用板子：前序+中序输出层序，比前序+中序输出先序增加一个变量index表示根结点在二叉树中的下标
·若二叉树的结点按层序遍历顺序存在数组中(根结点下标从0开始)，已知根结点下标为i，则以当前根结点为子树的左右孩子下标分别为2*i+1和2*i+2
· 镜面反转只需改变左右孩子index的值，左孩子为2 * index + 2， 右孩子为2 * index + 1
· map根据键值从小到大自动排序，递归完成后ans中的值就是层序遍历的顺序

#include<bits/stdc++.h>
using namespace std;
vector<int> pre,in;
map<int,int> ans;
void level(int root,int start,int end,int index)
	if(start>end) return;
	int k;
	for(k=start;k<end;k++) if(in[k]==pre[root]) break;
	ans[index] = pre[root];
	level(root+1,start,k-1,2*index+2);
	level(root+1+k-start,k+1,end,2*index+1);

int main()
	ios::sync_with_stdio(false);
	cin.tie(0),cout.tie(0);
	int N,flag=0;
	cin >> N;
	pre.resize(N);
	in.resize(N);
	for(int i=0;i<N;i++) cin >> in[i];
	for(int i=0;i<N;i++) cin >> pre[i];
	level(0,0,N-1,0);
	for(int i=0;i<ans.size();i++)
		if(ans[i])
			if(!flag) flag = 1;
			else cout << " ";
			cout << ans[i];
			
	
	return 0;