L2-006 树的遍历 (25分)

树的遍历算法解析

最新推荐文章于 2024-03-26 12:11:19 发布

原创最新推荐文章于 2024-03-26 12:11:19 发布 · 135 阅读

0 ·

CC 4.0 BY-SA版权

团体程序设计天梯赛-练习集专栏收录该内容

28 篇文章

订阅专栏

本文深入探讨了树的遍历算法，通过具体实例讲解了后序遍历、中序遍历到前序遍历的转换过程。使用C++实现了一个树的遍历程序，详细解释了递归算法的工作原理，以及如何通过比较节点的层级和索引来优化遍历顺序。

L2-006 树的遍历 (25分)

AC代码

#include <iostream>
#include <cstdio>
#include <vector>
#include <algorithm>
using namespace std;
struct node{
    int id, index, level;
};
vector<node> post, in, pre;
void pre_order(int root, int startt, int endd, int index, int level){
    if (startt > endd) return;
    int i = startt;
    while (i < endd && in[i].id != post[root].id) i++;
    pre.push_back({post[root].id, index, level});
    pre_order(root+i-endd-1, startt, i-1, index*2+1, level+1);
    pre_order(root-1, i+1, endd, index*2+2, level+1);
}
bool cmp(node x, node y){
    if (x.level != y.level) return x.level < y.level;
    return x.index < y.index;
}
int main(){
    int n;
    scanf("%d", &n);
    post.resize(n);
    in.resize(n);
    for (int i = 0; i < n; i++)
        scanf("%d", &post[i].id);
    for (int i = 0; i < n; i++)
        scanf("%d", &in[i].id);
    pre_order(n-1, 0, n-1, 0, 0);
    sort(pre.begin(), pre.end(), cmp);
    for (int i = 0; i < n; i++){
        if (i) printf(" ");
        printf("%d", pre[i].id);
    }
    return 0;
}