c_cpp 排序数组到平衡BST

Posted 2021-05-14

// https://www.geeksforgeeks.org/sorted-array-to-balanced-bst/
#include <iostream>
using namespace std;

struct node {
    int data;
    struct node* left;
    struct node* right;
};

node* bst (int a[], int l, int r) {
    if (l>r)
        return NULL;

    int m= (l+r)/2;
    node *head= (node*)malloc(sizeof(node));
    head->data= a[m];
    head->left = bst(a, l, m-1);
    head->right = bst(a, m+1, r);

    return head;
}

void preOrder(node *head) {
    if (head==NULL)
        return;
    cout<< head->data << " ";
    preOrder(head->left);
    preOrder(head->right);
}

int main() {
    int t;
    cin>>t;
    while (t-->0) {
        int n;
        cin>>n;
        int a[n];
        for (int i=0;i<n;i++)
            cin>>a[i];

        node* head= bst(a,0, n-1);
        preOrder(head);
    }
}