Update KMP algorithm

jeantimex · Oct 20, 2018 · b0adfa3 · b0adfa3
1 parent 6650633
commit b0adfa3
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/src/string/implement-strstr.js b/src/string/implement-strstr.js
@@ -81,16 +81,17 @@ const strStrKMP = (haystack, needle) => {
  */
 const getLPS = s => {
   const lps = Array(s.length).fill(0);
+
   let i = 1;
   let len = 0;
 
   while (i < s.length) {
     if (s[i] === s[len]) {
       lps[i++] = ++len;
-    } else if (len === 0) {
-      lps[i++] = 0;
-    } else {
+    } else if (len > 0) {
       len = lps[len - 1];
+    } else {
+      i++;
     }
   }