Tree Traversal [java]: Move main method into Tree class

Author: stormofice

contents/tree_traversal/code/java/MainClass.java

@@ -124,4 +124,34 @@ public int compareTo(Node other) {
             return Integer.compare(this.id, other.id);
         }
     }
+    
+    public static void main(String[] args) {
+        System.out.println("Creating Tree");
+        Tree tree = new Tree(3, 3);
+
+        System.out.println("Using recursive DFS :");
+        tree.dfsRecursive();
+
+        System.out.println("Using stack-based DFS :");
+        tree.dfsStack();
+
+        System.out.println("Using queue-based BFS :");
+        tree.bfsQueue();
+
+        System.out.println("Using post-order recursive DFS :");
+        tree.dfsRecursivePostOrder();
+
+
+        // Uncommenting the following 2 lines will result in an exception thrown because at least one Node of the Tree has more than 2 children and therefor a DFSRecursiveInorderBinary doesn't work.
+        System.out.println("Using in-order binary recursive DFS : (fail)");
+        tree.dfsRecursiveInOrderBinary();
+
+        tree = new Tree(3, 2);
+        System.out.println("Using in-order binary recursive DFS : (succeed)");
+        tree.dfsRecursiveInOrderBinary();
+
+
+        System.out.println("");
+    }
+    
 }

contents/tree_traversal/code/java/Tree.java

@@ -330,8 +330,6 @@ Here is a video describing tree traversal:
 {% sample lang="java" %}
 ##### Tree.java
 [import, lang:"java"](code/java/Tree.java)
-##### MainClass.java
-[import, lang:"java"](code/java/MainClass.java)
 {% sample lang="js" %}
 [import, lang:"javascript"](code/javascript/tree.js)
 {% sample lang="py" %}