Commitc9618c0

committed

add a solution for 841

1 parent2793ac5 commitc9618c0Copy full SHA for c9618c0

File tree

+28

-0

lines changed

src
- main/java/com/fishercoder/solutions
  - _841.java
- test/java/com/fishercoder
  - _841Test.java

+28

-0

lines changed

Lines changed: 24 additions & 0 deletions

Original file line number	Diff line number	Diff line change
`@@ -5,6 +5,7 @@`
`5`	`5`	`importjava.util.List;`
`6`	`6`	`importjava.util.Queue;`
`7`	`7`	`importjava.util.Set;`
	`8`	`+importjava.util.TreeSet;`
`8`	`9`
`9`	`10`	`publicclass_841 {`
`10`	`11`	`publicstaticclassSolution1 {`
`@@ -34,4 +35,27 @@ public boolean canVisitAllRooms(List<List<Integer>> rooms) {`
`34`	`35`	`returnunvisitedRooms.isEmpty();`
`35`	`36`	`}`
`36`	`37`	`}`
	`38`	`+`
	`39`	`+publicstaticclassSolution2 {`
	`40`	`+publicbooleancanVisitAllRooms(List<List<Integer>>rooms) {`
	`41`	`+TreeSet<Integer>treeSet =newTreeSet<>();`
	`42`	`+Set<Integer>visited =newHashSet<>();`
	`43`	`+visited.add(0);`
	`44`	`+treeSet.addAll(rooms.get(0));`
	`45`	`+while (!treeSet.isEmpty()) {`
	`46`	`+Integerkey =treeSet.pollFirst();`
	`47`	`+if (!visited.add(key)) {`
	`48`	`+continue;`
	`49`	`+ }`
	`50`	`+if (visited.size() ==rooms.size()) {`
	`51`	`+returntrue;`
	`52`	`+ }`
	`53`	`+treeSet.addAll(rooms.get(key));`
	`54`	`+ }`
	`55`	`+if (visited.size() ==rooms.size()) {`
	`56`	`+returntrue;`
	`57`	`+ }`
	`58`	`+returnfalse;`
	`59`	`+ }`
	`60`	`+ }`
`37`	`61`	`}`

Lines changed: 4 additions & 0 deletions

Original file line number	Diff line number	Diff line change
`@@ -12,11 +12,13 @@`
`12`	`12`
`13`	`13`	`publicclass_841Test {`
`14`	`14`	`privatestatic_841.Solution1solution1;`
	`15`	`+privatestatic_841.Solution2solution2;`
`15`	`16`	`privatestaticList<List<Integer>>rooms;`
`16`	`17`
`17`	`18`	`@BeforeClass`
`18`	`19`	`publicstaticvoidsetup() {`
`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`	`}`

Comments

(0)