【算法】二叉树顺序存储结构建树方法

最新推荐文章于 2023-10-06 15:40:30 发布

原创最新推荐文章于 2023-10-06 15:40:30 发布 · 1.1k 阅读

4 ·

CC 4.0 BY-SA版权

文章标签：

#二叉树 #算法 #数据结构

笔记专栏收录该内容

13 篇文章

订阅专栏

本文介绍了一种使用顺序存储结构构建二叉树的方法，通过动态数组存储节点，并利用递归函数创建二叉树。文章详细展示了如何定义二叉树的结构体，以及如何通过递归方式构建和遍历树，最后通过先序遍历验证了树的正确构建。

顺序存储结构，即使用数组存储，结点从下标为1的位置开始存储，a[i * 2]和a[i * 2 + 1]分别表示a[i]结点的左右孩子。其中若孩子为空，则用 -1 代替。

定义结构体：

struct node {
    int data;
    node *left, *right;
    node(int x) : data(x) { left = right = NULL; }
};

递归建树：（动态数组 a 为引用类型）

node* create(vector<int>& a, int i) {
    if (i >= a.size() || a[i] == -1) return NULL;
    node* root = new node(a[i]);
    root->left = create(a, i * 2);
    root->right = create(a, i * 2 + 1);
    return root;
}

完整代码：

#include <iostream>
#include <vector>
using namespace std;
struct node {
    int data;
    node *left, *right;
    node(int x) : data(x) { left = right = NULL; }
};
node* create(vector<int>& a, int i) {
    if (i >= a.size() || a[i] == -1) return NULL;
    node* root = new node(a[i]);
    root->left = create(a, i * 2);
    root->right = create(a, i * 2 + 1);
    return root;
}
/**
 * 先序遍历验证结果
 */
void preOrder(node* root) {
    if (root == NULL) return;
    printf("%d ", root->data);
    preOrder(root->left);
    preOrder(root->right);
}
int main() {
    vector<int> a;
    int temp[] = {-1, 5, 1, 4, -1, -1, 3, 6};
    for (int i = 0; i < 8; i++) a.push_back(temp[i]);
    node* root = create(a, 1);
    preOrder(root);	// 5 1 4 3 6
    return 0;
}