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

顺序存储结构，即使用数组存储，结点从下标为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;
}