refactor 139

fishercoder1534 · fishercoder1534 · commit 5a41238b4917 · 2018-04-28T07:54:25.000-07:00
diff --git a/src/main/java/com/fishercoder/solutions/_139.java b/src/main/java/com/fishercoder/solutions/_139.java
@@ -1,7 +1,6 @@
 package com.fishercoder.solutions;
 
 import java.util.List;
-import java.util.Set;
 
 /**
  * 139. Word Break
@@ -24,7 +23,7 @@ The wordDict parameter had been changed to a list of strings (instead of a set o
 public class _139 {
 
     public static class Solution1 {
-        /**this beats 70.46% submission. */
+        /** this beats 70.46% submission. */
         public boolean wordBreak(String s, List<String> wordDict) {
             int n = s.length();
             boolean[] dp = new boolean[n + 1];
@@ -75,7 +74,7 @@ public static class Solution3 {
          * Added pruning, plus start from the end to check.
          * This beats 95.20% submissions.
          */
-        public boolean wordBreak(String s, Set<String> wordDict) {
+        public boolean wordBreak(String s, List<String> wordDict) {
             int maxLen = Integer.MIN_VALUE;
             for (String word : wordDict) {
                 maxLen = (word.length() > maxLen) ? word.length() : maxLen;
@@ -85,7 +84,8 @@ public boolean wordBreak(String s, Set<String> wordDict) {
             boolean[] dp = new boolean[n + 1];
             dp[0] = true;
             for (int i = 1; i <= n; i++) {
-                for (int lastWordLength = 1; lastWordLength <= i && lastWordLength <= maxLen; lastWordLength++) {
+                for (int lastWordLength = 1; lastWordLength <= i && lastWordLength <= maxLen;
+                    lastWordLength++) {
                     if (!dp[i - lastWordLength]) {
                         continue;
                     }
diff --git a/src/test/java/com/fishercoder/_139Test.java b/src/test/java/com/fishercoder/_139Test.java
@@ -11,19 +11,25 @@
 import static junit.framework.Assert.assertEquals;
 
 public class _139Test {
-    private static _139.Solution2 solution2;
-    private static String s;
-    private static List<String> wordDict;
+  private static _139.Solution1 solution1;
+  private static _139.Solution2 solution2;
+  private static _139.Solution3 solution3;
+  private static String s;
+  private static List<String> wordDict;
 
-    @BeforeClass
-    public static void setup() {
-        solution2 = new _139.Solution2();
-    }
+  @BeforeClass
+  public static void setup() {
+    solution1 = new _139.Solution1();
+    solution2 = new _139.Solution2();
+    solution3 = new _139.Solution3();
+  }
 
-    @Test
-    public void test1() {
-        s = "leetcode";
-        wordDict = new ArrayList<>(Arrays.asList("leet", "code"));
-        assertEquals(true, solution2.wordBreak(s, wordDict));
-    }
+  @Test
+  public void test1() {
+    s = "leetcode";
+    wordDict = new ArrayList<>(Arrays.asList("leet", "code"));
+    assertEquals(true, solution1.wordBreak(s, wordDict));
+    assertEquals(true, solution2.wordBreak(s, wordDict));
+    assertEquals(true, solution3.wordBreak(s, wordDict));
+  }
 }
