This commit is contained in:
Weber 2024-04-27 19:18:24 +00:00
parent 3eb5131377
commit ac9e6f05d9

View File

@ -1,7 +1,5 @@
#include "binary_search_tree.h"
#include <stdlib.h>
#include <string.h>
#include <stdio.h>
static node_t *create_node(int data) {
node_t *new_node = malloc(sizeof(node_t));
@ -25,7 +23,6 @@ static node_t *insert_node(node_t *root, int data) {
return root;
}
node_t *build_tree(int *tree_data, size_t tree_data_len) {
node_t *root = NULL;
for (size_t i = 0; i < tree_data_len; i++) {
@ -34,7 +31,6 @@ node_t *build_tree(int *tree_data, size_t tree_data_len) {
return root;
}
void free_tree(node_t *tree) {
if (tree != NULL) {
free_tree(tree->left);
@ -43,7 +39,6 @@ void free_tree(node_t *tree) {
}
}
static void inorder_traversal(node_t *root, int *sorted_array, size_t *index) {
if (root != NULL) {
inorder_traversal(root->left, sorted_array, index);
@ -52,12 +47,24 @@ static void inorder_traversal(node_t *root, int *sorted_array, size_t *index) {
}
}
static size_t count_nodes_helper(node_t *root) {
if (root == NULL) {
return 0;
}
return 1 + count_nodes_helper(root->left) + count_nodes_helper(root->right);
}
size_t count_nodes(node_t *tree) {
return count_nodes_helper(tree);
}
int *sorted_data(node_t *tree) {
size_t index = 0;
int *sorted_array = malloc(sizeof(int) * count_nodes(tree));
size_t num_nodes = count_nodes(tree);
int *sorted_array = malloc(sizeof(int) * num_nodes);
if (sorted_array == NULL) {
return NULL;
return NULL;
}
inorder_traversal(tree, sorted_array, &index);
return sorted_array;
}
}