Clone Graph: AC

Author: sofkyle

LeetCodePrj/Java/leetcode/medium/page2/CloneGraph.java

@@ -0,0 +1,56 @@
+package leetcode.medium.page2;
+
+import java.util.*;
+
+public class CloneGraph {
+    static class Node {
+        public int val;
+        public List<Node> neighbors;
+
+        public Node() {
+            val = 0;
+            neighbors = new ArrayList<Node>();
+        }
+
+        public Node(int _val) {
+            val = _val;
+            neighbors = new ArrayList<Node>();
+        }
+
+        public Node(int _val, ArrayList<Node> _neighbors) {
+            val = _val;
+            neighbors = _neighbors;
+        }
+    }
+
+    public Node cloneGraph(Node node) {
+        if (node == null) {
+            return null;
+        }
+
+        Map<Node, Node> visited = new HashMap<>();
+        Queue<Node> queue = new LinkedList<>();
+        queue.offer(node);
+        Node res = new Node(node.val);
+        visited.put(node, res);
+
+        while (!queue.isEmpty()) {
+            Node oldNode = queue.poll();
+            Node newNode = visited.getOrDefault(oldNode, new Node(oldNode.val));
+
+            for (Node oldSubNode : oldNode.neighbors) {
+                Node newSubNode = null;
+                if (visited.containsKey(oldSubNode)) {
+                    newSubNode = visited.get(oldSubNode);
+                } else {
+                    queue.offer(oldSubNode);
+                    newSubNode = new Node(oldSubNode.val);
+                    visited.put(oldSubNode, newSubNode);
+                }
+                newNode.neighbors.add(newSubNode);
+            }
+        }
+
+        return res;
+    }
+}

LeetCodePrj/Java/leetcode/medium/page2/MediumPage2.java

@@ -1,9 +1,10 @@
 package leetcode.medium.page2;
 
 import leetcode.common.ListNode;
-import leetcode.hard.page2.LFUCache;
 import org.junit.jupiter.api.Test;
 
+import java.util.ArrayList;
+
 public class MediumPage2 {
     @Test
     public void testSortList() {
@@ -23,4 +24,12 @@ public void testEvalRPN() {
         EvaluateReversePolishNotation evalRPN = new EvaluateReversePolishNotation();
         System.out.println(evalRPN.evalRPN(input));
     }
+
+    @Test
+    public void testCloneMap() {
+        CloneGraph.Node node = new CloneGraph.Node();
+        node.neighbors = new ArrayList<>();
+        CloneGraph cloneGraph = new CloneGraph();
+        System.out.println(cloneGraph.cloneGraph(node));
+    }
 }