团体程序设计天梯赛L2-011 玩转二叉树

L2-011. 玩转二叉树

时间限制

400 ms

内存限制

65536 kB

代码长度限制

8000 B

判题程序

Standard

作者

陈越

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

输入格式：

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

输出格式：

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

输入样例：

71 2 3 4 5 6 74 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;}