Complete treeMap.js

treeMap/treeMap.js

@@ -23,8 +23,8 @@
   *  newTree.value // 2
   *  newTree.children[0].value // 4
   *  newTree.children[1].value // 6
-  *  newTree.children[0].children[1] // 10
-  *  newTree.children[1].children[1] // 14
+  *  newTree.children[0].children[1].value // 10
+  *  newTree.children[1].children[1].value // 14
   *  root1.value // still 1
   *
   *
@@ -38,13 +38,35 @@
  * Basic tree that stores a value.
 */
 
-var Tree = function(value){
+var Tree = function (value){
   this.value = value;
   this.children = [];
 };
 
 Tree.prototype.map = function (callback) {
-  // return a new tree with the same structure as `this`, with values generated by the callback
+  // create a new tree with value equal to after being modified by callback
+  var newTree = new Tree ( callback.call(null, this.value) );
+
+  // for each child in the original tree
+  this.children.forEach(function (child) {
+    // call child.map with the same callback
+    var result = child.map( callback );
+    // append the result of this call to new tree
+    newTree.addChild( result );
+  });
+
+  // return new tree
+  return newTree;
+};
+
+Tree.prototype.mapInPlace = function (callback) {
+  this.value = callback.call( null, this.value );
+
+  // for each child in tree
+  this.children.forEach(function (child) {
+    // call child.map with the same callback
+    child.mapInPlace( callback );
+  });
 };
 
 /**