ref: 095fdd12f103a99f1b5f247d9598c50e434c0478
parent: dcb14ae7ca299a2df34aea904c2dd1e6c43fb59b
author: qwx <[email protected]>
date: Sat Aug 15 21:39:50 EDT 2020
add kmp exact string search
--- /dev/null
+++ b/kmp.c
@@ -1,0 +1,50 @@
+#include <u.h>
+#include <libc.h>
+#include "asif.h"
+
+/* ordinary knuth-morris-pratt exact string search of a word W within a text S
+ * with W preprocessing */
+
+int *
+jumptab(String W)
+{+ int i, j, *T;
+
+ T = emalloc(W.n + 1);
+ T[0] = -1;
+ for(i=-1, j=0; j<W.n;){+ while(i > -1 && W.s[i] != W.s[j])
+ i = T[i];
+ i++;
+ j++;
+ if(W.s[i] == W.s[j])
+ T[j] = T[i];
+ else
+ T[j] = i;
+ }
+ return T;
+}
+
+VArray *
+kmpstrfind(String S, String W)
+{+ int n, i, j, *T;
+ VArray *v;
+
+ if(S.n < W.n)
+ return nil;
+ v = valloc(n, sizeof(int));
+ T = jumptab(W);
+ for(i=j=0; j<W.n;){+ while(i > -1 && W.s[i] != S.s[j])
+ i = T[i];
+ i++, j++;
+ if(i >= W.n){+ n = j - i;
+ vinsert(v, (void*)&n);
+ i = T[i];
+ }
+ }
+ free(T);
+ return v;
+}