L2-011. 玩转二叉树

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

---

提交代码

分析：建完树之后直接在bfs的时候左右调换位置。

#include <bits/stdc++.h>

using namespace std;
int n;
const int maxn=35;
int back[maxn];
int in[maxn];
struct node
{
    int data;
    node* lchild;
    node* rchild;
};
node* solve(int inl,int inr,int backl,int backr)
{
    if(inl>inr || backl>backr) return NULL;
    node* root=new node;
    int gen=back[backl];
    root->data=gen;
    int record;
    for(int i=inl;i<=inr;i++)
    {
        if(in[i]==gen)
        {
            record=i;
            break;
        }
    }
    int numleft=record-inl;
    root->lchild=solve(inl,record-1,backl+1,backl+numleft);
    root->rchild=solve(record+1,inr,backl+numleft+1,backr);
    return root;
}
vector<int> ans;
void bfs(node* root)
{
    queue<node*> q;
    q.push(root);
    while(!q.empty())
    {
        node* now=q.front();
        q.pop();
        ans.push_back(now->data);
        if(now->rchild!=NULL) q.push(now->rchild);
        if(now->lchild!=NULL) q.push(now->lchild);

    }
}
void print()
{
    for(int i=0;i<ans.size();i++)
    {
        cout<<ans[i];
        if(i!=ans.size()-1) printf(" ");
    }
}
int main()
{
    cin>>n;
    for(int i=0;i<n;i++)
    {
        scanf("%d",&in[i]);
    }
    for(int i=0;i<n;i++)
    {
        scanf("%d",&back[i]);
    }
    bfs(solve(0,n-1,0,n-1));
    print();


    return 0;
}