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 <bits/stdc++.h>using namespace std;struct node{    int data;    struct node *l, *r;};int a[1000];int cnt;int pre[1100],in[1100];struct node *creat(int *pre, int *in, int n){    for(int i = 0; i < n; i++)    {        if(pre[0] == in[i])        {            struct node *t = new node;            t->data = pre[0];            t->l = creat(pre+1, in, i);            t->r = creat(pre+i+1, in+i+1, n-(i+1));            return t;        }    }    return NULL;};void bfs(struct node *root){    queue<node*>q;    q.push(root);    while(!q.empty())    {        node *now = q.front();        q.pop();        //printf("%d ",now->data);        a[cnt++] = now->data;        if(now->r != NULL)        q.push(now->r);        if(now->l != NULL)        q.push(now->l);    }}int main(){    int n;    scanf("%d", &n);    for(int i = 0; i < n; i++)    {        scanf("%d", &in[i]);    }    for(int i = 0; i < n; i++)        scanf("%d", &pre[i]);        struct node *root = creat(pre, in, n);        cnt = 0;        bfs(root);        for(int i = 0; i <cnt - 1; i++)            printf("%d ",a[i]);        printf("%d\n",a[cnt-1]);}