word break solution with pruning 2d

Author: yhkee0404

word-break/yhkee0404.scala

@@ -4,15 +4,13 @@ object Solution {
         dp(0) = true
         (0 to s.length - 1).exists { i =>
             if (! dp(i)) false
-            else {
-                for (word <- wordDict if ! dp(s.length)) {
-                    val j = i + word.length
-                    if (j <= s.length && ! dp(j) && s.substring(i, j) == word) {
-                        dp(j) = true // T(s, wordDict, word) = O(s.length * wordDict.length * word.length)
+            else wordDict.exists { word =>
+                        val j = i + word.length
+                        if (j <= s.length && ! dp(j) && s.substring(i, j) == word) {
+                            dp(j) = true // T(s, wordDict, word) = O(s.length * wordDict.length * word.length)
+                        }
+                        dp(s.length)
                     }
-                }
-                dp(s.length)
-            }
         }
     }
 }