add a solution for 841

fishercoder1534 · fishercoder1534 · commit c9618c00b121 · 2021-03-19T07:48:59.000-07:00
diff --git a/src/main/java/com/fishercoder/solutions/_841.java b/src/main/java/com/fishercoder/solutions/_841.java
@@ -5,6 +5,7 @@
 import java.util.List;
 import java.util.Queue;
 import java.util.Set;
+import java.util.TreeSet;
 
 public class _841 {
     public static class Solution1 {
@@ -34,4 +35,27 @@ public boolean canVisitAllRooms(List<List<Integer>> rooms) {
             return unvisitedRooms.isEmpty();
         }
     }
+
+    public static class Solution2 {
+        public boolean canVisitAllRooms(List<List<Integer>> rooms) {
+            TreeSet<Integer> treeSet = new TreeSet<>();
+            Set<Integer> visited = new HashSet<>();
+            visited.add(0);
+            treeSet.addAll(rooms.get(0));
+            while (!treeSet.isEmpty()) {
+                Integer key = treeSet.pollFirst();
+                if (!visited.add(key)) {
+                    continue;
+                }
+                if (visited.size() == rooms.size()) {
+                    return true;
+                }
+                treeSet.addAll(rooms.get(key));
+            }
+            if (visited.size() == rooms.size()) {
+                return true;
+            }
+            return false;
+        }
+    }
 }
diff --git a/src/test/java/com/fishercoder/_841Test.java b/src/test/java/com/fishercoder/_841Test.java
@@ -12,11 +12,13 @@
 
 public class _841Test {
     private static _841.Solution1 solution1;
+    private static _841.Solution2 solution2;
     private static List<List<Integer>> rooms;
 
     @BeforeClass
     public static void setup() {
         solution1 = new _841.Solution1();
+        solution2 = new _841.Solution2();
     }
 
     @Test
@@ -27,6 +29,7 @@ public void test1() {
         rooms.add(Arrays.asList(3));
         rooms.add(Arrays.asList());
         assertEquals(true, solution1.canVisitAllRooms(rooms));
+        assertEquals(true, solution2.canVisitAllRooms(rooms));
     }
 
     @Test
@@ -37,5 +40,6 @@ public void test2() {
         rooms.add(Arrays.asList(2));
         rooms.add(Arrays.asList(0));
         assertEquals(false, solution1.canVisitAllRooms(rooms));
+        assertEquals(false, solution2.canVisitAllRooms(rooms));
     }
 }

Original file line number	Diff line number	Diff line change
`@@ -12,11 +12,13 @@`
`12`	`12`
`13`	`13`	`public class _841Test {`
`14`	`14`	`private static _841.Solution1 solution1;`
	`15`	`+ private static _841.Solution2 solution2;`
`15`	`16`	`private static List<List<Integer>> rooms;`
`16`	`17`
`17`	`18`	`@BeforeClass`
`18`	`19`	`public static void setup() {`
`19`	`20`	`solution1 = new _841.Solution1();`
	`21`	`+ solution2 = new _841.Solution2();`
`20`	`22`	`}`
`21`	`23`
`22`	`24`	`@Test`
`@@ -27,6 +29,7 @@ public void test1() {`
`27`	`29`	`rooms.add(Arrays.asList(3));`
`28`	`30`	`rooms.add(Arrays.asList());`
`29`	`31`	`assertEquals(true, solution1.canVisitAllRooms(rooms));`
	`32`	`+ assertEquals(true, solution2.canVisitAllRooms(rooms));`
`30`	`33`	`}`
`31`	`34`
`32`	`35`	`@Test`
`@@ -37,5 +40,6 @@ public void test2() {`
`37`	`40`	`rooms.add(Arrays.asList(2));`
`38`	`41`	`rooms.add(Arrays.asList(0));`
`39`	`42`	`assertEquals(false, solution1.canVisitAllRooms(rooms));`
	`43`	`+ assertEquals(false, solution2.canVisitAllRooms(rooms));`
`40`	`44`	`}`
`41`	`45`	`}`