hello-algo/en/codes/c/chapter_tree/binary_tree.c

44 lines
944 B
C

/**
* File: binary_tree.c
* Created Time: 2023-01-11
* Author: Reanon (793584285@qq.com)
*/
#include "../utils/common.h"
/* Driver Code */
int main() {
/* Initialize binary tree */
// Initialize node
TreeNode *n1 = newTreeNode(1);
TreeNode *n2 = newTreeNode(2);
TreeNode *n3 = newTreeNode(3);
TreeNode *n4 = newTreeNode(4);
TreeNode *n5 = newTreeNode(5);
// Construct node references (pointers)
n1->left = n2;
n1->right = n3;
n2->left = n4;
n2->right = n5;
printf("Initialize binary tree\n");
printTree(n1);
/* Insert and remove nodes */
TreeNode *P = newTreeNode(0);
// Insert node P between n1 -> n2
n1->left = P;
P->left = n2;
printf("After inserting node P\n");
printTree(n1);
// Remove node P
n1->left = n2;
// Free memory
free(P);
printf("After removing node P\n");
printTree(n1);
freeMemoryTree(n1);
return 0;
}