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

Here_SDUT

发布于 2022-08-08 16:27:54

3130

发布于 2022-08-08 16:27:54

文章被收录于专栏：机器学习炼丹之旅

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

输入格式： 输入第一行给出一个正整数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

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

代码

//                              _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;
}

本文参与腾讯云自媒体同步曝光计划，分享自作者个人站点/博客。

原始发表：2021-3-19 1，如有侵权请联系 cloudcommunity@tencent.com 删除

编程算法