clone-graph solution

Author: lhc0506

clone-graph/lhc0506.js

@@ -0,0 +1,36 @@
+/**
+ * // Definition for a _Node.
+ * function _Node(val, neighbors) {
+ *    this.val = val === undefined ? 0 : val;
+ *    this.neighbors = neighbors === undefined ? [] : neighbors;
+ * };
+ */
+
+/**
+ * @param {_Node} node
+ * @return {_Node}
+ */
+var cloneGraph = function(node) {
+  if (!node) return null;
+
+  const root = new Node(node.val);
+  const stack = [node];
+  const map = new Map();
+
+  map.set(node.val, new _Node(node.val));
+
+  while (stack.length > 0) {
+      const currentNode = stack.pop();
+      const newCurrentNode = new Node(currentNode.val);
+
+      for (const neighbor of currentNode.neighbors) {
+          if (!map.has(neighbor.val)) {
+              map.set(neighbor.val, new Node(neighbor.val));
+              stack.push(neighbor);
+          }
+          map.get(currentNode.val).neighbors.push(map.get(neighbor.val));
+      }
+
+  }
+  return map.get(node.val);
+};