pat 团体天梯赛 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
 
思路：先依据前序和中序遍历重建二叉树，递归的对换左右孩子，最后中序遍历输出结果即可。
AC代码：

#define _CRT_SECURE_NO_DEPRECATE
#include<iostream>
#include<cstdio>
#include<vector>
#include<algorithm>
#include<cstring>
#include<string>
#include<queue>
#include<cmath>
#include<map>
#include<set>
using namespace std;
#define INF 0x3f3f3f3f
#define EPS 1e-5
#define pi cos(-1)
const int N_MAX =  + ;
 
struct Node {
    int r, l, key;
}node[N_MAX];
 
vector<int> in,pre,post;
 
int n,pos=;
void dfs(const int& l,const int &r,int n) {//找到后序遍历中区间[l,r]的根结点和左右子树
    if (l > r) {
        node[n].key = INF;
        return;
    }
    int root = post[pos++];
    node[n].l =  * n;
    node[n].r =  * n+;
    node[n].key = root;
    int m = find(in.begin(), in.end(), root) - in.begin();
    dfs(l, m - ,  * n);
    dfs(m + , r,  * n + );
}
 
void exchange(Node *node,int n) {//将所有非叶结点的左右孩子对换
    if (node[n].key == INF)return;
    if(node[n].l != )exchange(node,  * n);
    if(node[n].r != )exchange(node,  * n + );
    if (node[n].l !=  && node[n].r != ) {
        swap(node[n].l, node[n].r);//交换左右孩子
    }
}
 
void bfs() {
    queue<int>que;
    que.push();
    while (!que.empty()) {
        int p = que.front(); que.pop();
        if(node[p].key!=INF)pre.push_back(node[p].key);
        if (node[p].l != )que.push(node[p].l);
        if (node[p].r != )que.push(node[p].r);
    }
}
 
int main(){
    while (cin>>n) {
        pos = ;
        in.resize(n);
        post.resize(n);
        for (int i = ; i < n;i++) {
            cin >> in[i];
        }
        for (int i = ; i < n;i++) {
            cin >> post[i];
        }
        dfs(, n - , );
        exchange(node, );
        bfs();
        for (int i = ; i < n;i++) {
            cout << pre[i];
            if (i +  == n)cout << endl;
            else cout << " ";
        }
    }
    return ;
}