L2-011 玩转二叉树 (25 分) (二叉树)

2022-08-08 16:27:54 浏览数 (1)

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

输入格式: 输入第一行给出一个正整数N(≤30),是二叉树中结点的个数。第二行给出其中序遍历序列。第三行给出其前序遍历序列。数字间以空格分隔。

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

输入样例:

代码语言:javascript复制
7
1 2 3 4 5 6 7
4 1 3 2 6 5 7

输出样例:

代码语言:javascript复制
4 6 1 7 5 3 2

分析 就是两种遍历方式求另外一种,PTA好喜欢这种题,镜面翻转其实就是在求层序的时候反着放入队列就行。

代码

代码语言:javascript复制
//                              _ooOoo_
//                             o8888888o
//                             88" . "88
//                             (| -_- |)
//                              O = /O
//                           ____/`---'____
//                        .   ' | |// `.
//                         / ||| : |||// 
//                        / _||||| -:- |||||- 
//                         | | \ - /// | |
//                       | _| ''---/'' | |
//                         .-__ `-` ___/-. /
//                    ___`. .' /--.-- `. . __
//                  ."" '< `.____<|>_/___.' >'"".
//                 | | : `- `.;` _ /`;.`/ - ` : | |
//                      `-. _ __ /__ _/ .-` / /
//           ======`-.____`-.________/___.-`____.-'======
//                              `=---='
//
//           .............................................
//                   佛祖保佑     一发AC     永无BUG
#include <bits/stdc  .h>
#define LL long long
using namespace std;
const int maxn = 1e5   10;
const double PI = acos(-1.0);
int zhong[50], qian[50];
struct node {
    int data;
    node *l, *r;
};
int now = 0;
int n;
node *creat(int l, int r) {
    if (l > r) return NULL;
    int mid;
    node *root = new node;
    root->data = qian[now];
    for (int i = l; i <= r; i  ) {
        if (qian[now] == zhong[i]) {
            mid = i;
            break;
        }
    }
    now  ;
    root->l = creat(l, mid - 1);
    root->r = creat(mid   1, r);
    return root;
}

vector<int> ans;
void ceng(node *root) {
    queue<node *> q;
    q.push(root);
    while (q.size()) {
        auto now = q.front();
        q.pop();
        if (now == NULL) continue;
        ans.push_back(now->data);
        q.push(now->r);
        q.push(now->l);
    }
}

int main(int argc, char const *argv[]) {
    cin >> n;
    for (int i = 0; i < n; i  ) cin >> zhong[i];
    for (int i = 0; i < n; i  ) cin >> qian[i];
    node *root = creat(0, n - 1);
    ceng(root);
    for (int i = 0; i < ans.size(); i  ) {
        if (i == ans.size() - 1)
            cout << ans[i] << endl;
        else
            cout << ans[i] << ' ';
    }
    return 0;
}

0 人点赞