-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathbtree.c
87 lines (75 loc) · 1.68 KB
/
btree.c
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
#include <stdio.h>
#include <stdlib.h>
struct Node {
int val;
struct Node* left;
struct Node* right;
};
typedef struct {
struct Node *root;
int level;
} BinaryTree;
enum NodePosition {
LEFT,
RIGHT,
ROOT,
};
void insert(BinaryTree **tree, struct Node *node)
{
struct Node *prev;
if(tree == NULL || (*tree) == NULL || node == NULL) return;
struct Node **head = &(*tree)->root;
while(*head) {
prev = (*head);
if(node->val > (*head)->val) head = &((*head)->right);
else if(node->val < (*head)->val) head = &((*head)->left);
}
if(prev == NULL || ((prev->left == NULL) && (prev->right == NULL))) (*tree)->level++;
(*head) = node;
}
void printnode(struct Node *head, int level, enum NodePosition np)
{
if(head) {
const char *POS = ((np == LEFT) ? "LEFT: " : ((np == ROOT) ? "ROOT: " : "RIGHT: "));
printf("[%d] %s%.*s%d%.*s\n", level, POS, level + 1, "\t", head->val, level + 1, "\t");
if(head->left) printnode(head->left, level - 1, LEFT);
if(head->right) printnode(head->right, level - 1, RIGHT);
}
}
void printtree(BinaryTree *tree)
{
printf("Binary Tree\n");
printf("levels: %d\n\n", tree->level);
printnode(tree->root, tree->level, ROOT);
}
int main(void)
{
BinaryTree *tree;
if(!(tree = (BinaryTree *) malloc(sizeof(BinaryTree)))) return 1;
struct Node root = {
.val = 5,
.left = NULL,
.right = NULL,
};
struct Node left = {
.val = 2,
.left = NULL,
.right = NULL,
};
struct Node right = {
.val = 9,
.left = NULL,
.right = NULL,
};
struct Node right_again = {
.val = 12,
.left = NULL,
.right = NULL,
};
insert(&tree, &root);
insert(&tree, &left);
insert(&tree, &right);
insert(&tree, &right_again);
printtree(tree);
free(tree);
}