pvjc24/a3/binary_search_tree.c

76 lines
1.6 KiB
C
Raw Normal View History

2024-04-18 12:47:26 +00:00
#ifndef BINARY_SEARCH_TREE_H
#define BINARY_SEARCH_TREE_H
#include <stddef.h>
2024-04-18 12:44:33 +00:00
2024-04-18 12:47:26 +00:00
typedef struct node node_t;
2024-04-18 12:44:33 +00:00
2024-04-18 12:47:26 +00:00
struct node {
node_t *right;
node_t *left;
int data;
};
2024-04-18 12:44:33 +00:00
2024-04-18 12:47:26 +00:00
node_t *build_tree(int *tree_data, size_t tree_data_len);
2024-04-18 12:44:33 +00:00
void free_tree(node_t *tree);
2024-04-18 12:47:26 +00:00
int *sorted_data(node_t *tree);
2024-04-18 12:44:33 +00:00
2024-04-18 12:47:26 +00:00
#endif
#include "binary_search_tree.h"
#include <stdlib.h>
2024-04-18 12:44:33 +00:00
2024-04-18 12:47:26 +00:00
node_t *new_node(int value) {
node_t *node = malloc(sizeof(node_t));
if (node != NULL) {
node->data = value;
node->left = NULL;
node->right = NULL;
2024-04-18 12:44:33 +00:00
}
2024-04-18 12:47:26 +00:00
return node;
}
node_t *insert_node(node_t *root, int value) {
if (root == NULL)
return new_node(value);
if (value <= root->data)
root->left = insert_node(root->left, value);
else
root->right = insert_node(root->right, value);
2024-04-18 12:44:33 +00:00
return root;
}
2024-04-18 12:47:26 +00:00
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++) {
root = insert_node(root, tree_data[i]);
}
return root;
}
2024-04-18 12:44:33 +00:00
void free_tree(node_t *tree) {
if (tree != NULL) {
free_tree(tree->left);
free_tree(tree->right);
free(tree);
}
}
2024-04-18 12:47:26 +00:00
void tree_to_sorted_array(node_t *tree, int arr[], int *index) {
2024-04-18 12:44:33 +00:00
if (tree != NULL) {
2024-04-18 12:47:26 +00:00
tree_to_sorted_array(tree->left, arr, index);
arr[(*index)++] = tree->data;
tree_to_sorted_array(tree->right, arr, index);
2024-04-18 12:44:33 +00:00
}
}
2024-04-18 12:47:26 +00:00
int *sorted_data(node_t *tree) {
if (tree == NULL)
2024-04-18 12:44:33 +00:00
return NULL;
2024-04-18 12:47:26 +00:00
int *arr = malloc(100 * sizeof(int));
tree_to_sorted_array(tree, arr, &index);
2024-04-18 12:44:33 +00:00
return arr;
}