L2-011. 玩转二叉树
时间限制
400 ms
内存限制
65536 kB
代码长度限制
8000 B
判题程序
Standard
作者
陈越
给定一棵二叉树的中序遍历和前序遍历,请你先将树做个镜面反转,再输出反转后的层序遍历的序列。所谓镜面反转,是指将所有非叶结点的左右孩子对换。这里假设键值都是互不相等的正整数。
输入格式:
输入第一行给出一个正整数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<stdio.h>
#include<algorithm>
#include<string.h>
#include<queue>
using namespace std;
const int N=10005;
const int mod=1e9+7;
int son[N][2];
int a[N],b[N],rt,n;
int dfs(int l,int r,int ll,int rr)
{
if(l>r||ll>rr) return 0;
for(int i=l; i<=r; i++)
{
if(b[ll]==a[i])
{
son[b[ll]][0]=dfs(l,i-1,ll+1,i-l+ll);
son[b[ll]][1]=dfs(i+1,r,i-l+ll+1,rr);
}
}
return b[ll];
}
int main()
{
while(~scanf("%d",&n))
{
for(int i=1; i<=n; i++) scanf("%d",&a[i]);
for(int i=1; i<=n; i++) scanf("%d",&b[i]);
dfs(1,n,1,n);
queue<int> q;
q.push(b[1]);
printf("%d",b[1]);
while(!q.empty())
{
int now=q.front();
q.pop();
if(son[now][1])
{
q.push(son[now][1]);
printf(" %d",son[now][1]);
}
if(son[now][0])
{
q.push(son[now][0]);
printf(" %d",son[now][0]);
}
}
printf("\n");
}
return 0;
}