Import an example for BinarySearchTree

1 files changed, 135 insertions, 0 deletions

diff --git a/src/example/java/bjc/funcdata/bst/BinarySearchTreeExample.java b/src/example/java/bjc/funcdata/bst/BinarySearchTreeExample.java
new file mode 100644
index 0000000..b5b09e3
--- /dev/null
+++ b/src/example/java/bjc/funcdata/bst/BinarySearchTreeExample.java
@@ -0,0 +1,135 @@
+package bjc.funcdata.bst;
+
+import java.io.*;
+import java.util.*;
+
+/**
+ * Example showing how to use the binary search tree.
+ *
+ * @author ben
+ *
+ */
+public class BinarySearchTreeExample {
+	private static void display(final BinarySearchTree<Character> tree,
+			final Scanner input, PrintStream output) {
+		output.print(
+				"What order would you like the tree to be printed in (m for options): ");
+		char command;
+
+		while (true) {
+			command = input.nextLine().charAt(0);
+			TreeLinearizationMethod method = null;
+
+			switch (command) {
+			case 'm':
+				System.out.println("Possible tree printing methods: ");
+				System.out.println(
+						"\tp: Preorder printing (print parent first, then left & right).");
+				System.out.println(
+						"\ti: Inorder printing (print left first, then parent & right).");
+				System.out.println(
+						"\to: Postorder printing (print left first, then right & parent).");
+				break;
+			case 'p':
+				method = TreeLinearizationMethod.PREORDER;
+				break;
+			case 'i':
+				method = TreeLinearizationMethod.INORDER;
+				break;
+			case 'o':
+				method = TreeLinearizationMethod.POSTORDER;
+				break;
+			default:
+				System.out.println("ERROR: Unknown command.");
+			}
+
+			if (method != null) {
+				tree.traverse(method, element -> {
+					System.out.println("Node: " + element);
+					return true;
+				});
+
+				return;
+			}
+
+			System.out.print(
+					"What order would you like the tree to be printed in (m for options): ");
+		}
+	}
+
+	/**
+	 * Main method of class
+	 *
+	 * @param args
+	 *             Unused CLI args
+	 */
+	public static void main(final String[] args) {
+		final Scanner input = new Scanner(System.in);
+		
+		runExample(input, System.out);
+
+		input.close();
+	}
+
+	private static void runExample(final Scanner input, OutputStream outpt) {
+		System.out.println("Binary Tree Constructor/Searcher");
+		final BinarySearchTree<Character> tree
+				= new BinarySearchTree<>((o1, o2) -> o1 - o2);
+
+		PrintStream output = new PrintStream(outpt);
+		
+		char command = ' ';
+		while (command != 'e') {
+			output.print("Enter a command (m for help): ");
+			command = input.nextLine().charAt(0);
+
+			switch (command) {
+			case 'm':
+				output.println("Valid commands: ");
+				output.println("\tm: Display this help message.");
+				output.println("\te: Exit this program.");
+				output.println("\ta: Add a node to the binary tree.");
+				output.println("\td: Display the binary tree.");
+				output.println("\tr: Remove a node from the binary tree.");
+				output.println("\tf: Check if a given node is in the binary tree.");
+				output.println("\tt: Trim all deleted nodes from the tree.");
+				output.println("\tb: Balance the tree (also trims dead nodes)");
+				break;
+			case 'a':
+				output.print("Enter the letter to add to the binary tree: ");
+				command = input.nextLine().charAt(0);
+
+				tree.addNode(command);
+				break;
+			case 'r':
+				output.print("Enter the letter to add to the binary tree: ");
+				command = input.nextLine().charAt(0);
+
+				tree.deleteNode(command);
+				break;
+			case 'd':
+				display(tree, input, output);
+				break;
+			case 'f':
+				output.print("Enter the letter to add to the binary tree: ");
+				command = input.nextLine().charAt(0);
+
+				final boolean inTree = tree.isInTree(command);
+				if (inTree) {
+					output.printf("Node %s was found\n", command);
+				} else {
+					output.printf("Node %s was not found\n", command);
+				}
+				break;
+			case 't':
+				tree.trim();
+				break;
+			case 'b':
+				tree.balance();
+				break;
+			default:
+				output.println("ERROR: Unrecognized command.");
+			}
+		}
+	}
+}