Delegate to the single traverse function

1 files changed, 2 insertions, 3 deletions

diff --git a/src/02/01/binary_tree.h b/src/02/01/binary_tree.h
index a8d5fb8..93a6f12 100644
--- a/src/02/01/binary_tree.h
+++ b/src/02/01/binary_tree.h
@@ -6,9 +6,8 @@ struct node {
 
 typedef struct node Node;
 typedef void(Visitor)(Node* node);
+enum Traversal { INORDER = 1, PREORDER = 2, POSTORDER = 4 };
 
 Node *initialize(int data);
-void preorder_traversal(Node *node, Visitor visitor);
-void inorder_traversal(Node *node, Visitor visitor);
-void postorder_traversal(Node *node, Visitor visitor);
+void traverse(Node *node, Visitor visitor, enum Traversal traversal);
 void destroy(Node *head);