Level Order Traversal for red black tree using queue.

shaunak1111 · shaunak1111 · commit 40b0658968be · 2017-03-22T15:06:30.000+05:30
diff --git a/src/data-structures/red-black-tree.js b/src/data-structures/red-black-tree.js
@@ -266,4 +266,34 @@
     }
   };
 
+  /**
+  * Get Level Order Traversal for the given Red Black Tree,
+  * returns 'Tree is empty' string when tree has no Nodes.
+  * Complexity: O(N).
+  *
+  * @public
+  * @return {String} The keys of the tree in level order traversal.
+  *
+  */
+  exports.RBTree.prototype.levelOrderTraversal = function () {
+    var queue = [];
+    var levelOrderString = '';
+    if (this._root){
+      queue.push(this._root);
+    }else {
+      levelOrderString = ' Tree is empty';
+    }
+    while (queue.length !== 0){
+      var tempNode = queue.shift();
+      levelOrderString += ' ' + tempNode.getKey();
+      if (tempNode.getLeft() !== null){
+        queue.push(tempNode.getLeft());
+      }
+      if (tempNode.getRight() !== null){
+        queue.push(tempNode.getRight());
+      }
+    }
+    return 'Level Order Traversal -:' + levelOrderString;
+  };
+
 })(typeof window === 'undefined' ? module.exports : window);
diff --git a/test/data-structures/red-black-tree.spec.js b/test/data-structures/red-black-tree.spec.js
@@ -97,4 +97,20 @@ describe('RBTree', function () {
     });
   });
 
+  describe('levelOrderTraversal method', function () {
+    it('should be able to traverse tree in level order', function () {
+      var tree = new RBTree();
+      expect(tree.levelOrderTraversal()).toBe('Level Order Traversal -: Tree is empty');
+      tree.put(10);
+      tree.put(20);
+      expect(tree.levelOrderTraversal()).toBe('Level Order Traversal -: 20 10');
+      tree.put(30);
+      expect(tree.levelOrderTraversal()).toBe('Level Order Traversal -: 20 10 30');
+      tree.put(45);
+      expect(tree.levelOrderTraversal()).toBe('Level Order Traversal -: 20 10 45 30');
+      tree.put(5);
+      expect(tree.levelOrderTraversal()).toBe('Level Order Traversal -: 20 10 45 5 30');
+    });
+  });
+
 });
