二维数组构建二叉树

最新推荐文章于 2023-03-10 15:23:36 发布

原创最新推荐文章于 2023-03-10 15:23:36 发布 · 3.1k 阅读

3 ·

CC 4.0 BY-SA版权

根据给定的先序和中序遍历数组，利用二维数组构建二叉树。介绍如何通过输入的两个整数数组，生成正确表示二叉树的后序数组，用0作为空节点。

二维数组构建二叉树

给出一棵二叉树的先序和中序数组，通过这两个数组直接生成正确的后序数组。
第一行一个整数 n，表示二叉树的大小。
第二行 n 个整数 a_i，表示二叉树的先序遍历数组。
第三行 n 个整数 b_i，表示二叉树的中序遍历数组。

这里使用二维数组来构建树，使用0作为树的空节点。

C++

#include <iostream>
#include <cstdio>
#include <cstring>
 
using namespace std;
 
const int MAXN = 10000 + 10;
int preorder[MAXN], inorder[MAXN], tree[MAXN][2];
int n;
 
int build(int idx, int l, int r)
{
    if(idx >= n) return 0;
    if(l > r) return 0;
    int root =  preorder[idx];
    int pos = l;
    while(pos <= r && inorder[pos] != root) pos ++;
    int cnt = pos - l;
    tree[root][0] = build(idx + 1, l, pos - 1);
    tree[root][1] = build(idx + cnt + 1, pos + 1, r);
    return root;
}
 
void postorder(int root){
    if(root == 0) return;
    postorder(tree[root][0]);
    postorder(tree[root][1]);
    printf("%d ", root);
}
 
int main()
{
    scanf("%d", &n);
    for(int i = 0; i < n; i ++)
        scanf("%d", &preorder[i]);
    for(int i = 0; i < n; i ++)
        scanf("%d", &inorder[i]);
    int ans = build(0, 0, n - 1);
    postorder(ans);
    return 0;
}