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 "binary_search_tree.h"
#include <stdlib.h> #include <stdlib.h>
#include <string.h>
#include <stdio.h>
static node_t *create_node(int data) { static node_t *create_node(int data) {
node_t *new_node = malloc(sizeof(node_t)); node_t *new_node = malloc(sizeof(node_t));
@ -25,7 +23,6 @@ static node_t *insert_node(node_t *root, int data) {
return root; return root;
} }
node_t *build_tree(int *tree_data, size_t tree_data_len) { node_t *build_tree(int *tree_data, size_t tree_data_len) {
node_t *root = NULL; node_t *root = NULL;
for (size_t i = 0; i < tree_data_len; i++) { 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; return root;
} }
void free_tree(node_t *tree) { void free_tree(node_t *tree) {
if (tree != NULL) { if (tree != NULL) {
free_tree(tree->left); 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) { static void inorder_traversal(node_t *root, int *sorted_array, size_t *index) {
if (root != NULL) { if (root != NULL) {
inorder_traversal(root->left, sorted_array, index); inorder_traversal(root->left, sorted_array, index);
@ -52,9 +47,21 @@ 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) { int *sorted_data(node_t *tree) {
size_t index = 0; 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) { if (sorted_array == NULL) {
return NULL; return NULL;
} }