Merge pull request TheAlgorithms#164 from KyleScharnhorst/master

Add: level order traversal.

Author: varunu28

data_structures/Trees/TreeTraversal.java

@@ -3,25 +3,33 @@
  * @author Varun Upadhyay (https://github.com/varunu28)
  *
  */
+import java.util.LinkedList;
 
 // Driver Program
 public class TreeTraversal {
     public static void main(String[] args) {
         Node tree = new Node(5);
         tree.insert(3);
         tree.insert(7);
+        tree.insert(1);
+        tree.insert(9);
 
-        // Prints 3 5 7
+        // Prints 1 3 5 7 9
         tree.printInOrder();
         System.out.println();
 
-        // Prints 5 3 7
+        // Prints 5 3 1 7 9
         tree.printPreOrder();
         System.out.println();
 
-        // Prints 3 7 5
+        // Prints 1 3 9 7 5
         tree.printPostOrder();
         System.out.println();
+
+        // Add a couple more nodes for print level test
+        // Print 5 3 7 1 9
+        tree.printLevelOrder();
+        System.out.println();
     }
 }
 
@@ -31,6 +39,7 @@ public static void main(String[] args) {
  * printInOrder: LEFT -> ROOT -> RIGHT
  * printPreOrder: ROOT -> LEFT -> RIGHT
  * printPostOrder: LEFT -> RIGHT -> ROOT
+ * printLevelOrder: ROOT -> ROOT's CHILDREN -> ROOT's CHILDREN's CHILDREN -> etc
  */
 class Node {
     Node left, right;
@@ -88,5 +97,20 @@ public void printPostOrder() {
         }
         System.out.print(data + " ");
     }
+
+    public void printLevelOrder() {
+        LinkedList<Node> queue = new LinkedList<>();
+        queue.add(this);
+        while(!queue.isEmpty()) {
+            Node n = queue.poll();
+            System.out.print(n.data + " ");
+            if (n.left != null) {
+                queue.add(n.left);
+            }
+            if (n.right != null) {
+                queue.add(n.right);
+            }
+        }
+    }
 }