题目:
给定一棵二叉树的中序遍历和前序遍历,请你先将树做个镜面反转,再输出反转后的层序遍历的序列。所谓镜面反转,是指将所有非叶结点的左右孩子对换。这里假设键值都是互不相等的正整数。
输入格式:
输入第一行给出一个正整数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
#include<iostream>
#include<vector>
#include<queue>
#include<cstring>
using namespace std;
int tree[10000];///树数组
void BulidTree(int index, vector<int>a, vector<int>b)///循环建树
{
if(a.size() == 0)return ;
int p = 0;
for(int i = 0; i < a.size(); i++)
{
if(b[0] == a[i])
{
p = i;
break;
}
}///在中序遍历中找到先序遍历的第一个数(根)
vector<int>al,bl,ar,br;///中序左部,先序左部,中序右部,先序右部
for(int i = 1; i <= p; i++)bl.push_back(b[i]);///先序根左部进入容器
for(int i = p+1; i < b.size(); i++)br.push_back(b[i]);///先序根右部进入容器
for(int i = 0; i < p; i++)al.push_back(a[i]);///中序根左部进入容器
for(int i = p + 1; i < a.size(); i++)ar.push_back(a[i]);///中序根右部进入容器
tree[index] = a[p];///树数组
BulidTree(index*2, ar, br);
BulidTree(index*2+1, al, bl);///根左右互换
}
void Print()///输出
{
queue<int>q;
int index = 1;
q.push(index);
int t = 0;
while(q.size())
{
if(t>0)cout<<" ";
else t++;
index = q.front();
cout<<tree[index];
q.pop();
if(tree[index*2] != -1)q.push(index*2);
if(tree[index*2+1] != -1)q.push(index*2+1);
}
return ;
}
int main()
{
int n, t;
memset(tree,-1,sizeof(tree));
vector<int>a,b;///中序,先序
cin>>n;
for(int i = 0; i < n; i++)
{
cin>>t;
a.push_back(t);
}
for(int i = 0; i < n; i++)
{
cin>>t;
b.push_back(t);
}
BulidTree(1,a,b);
Print();
return 0;
}