shithub: asif

ref: 095fdd12f103a99f1b5f247d9598c50e434c0478
dir: /kmp.c/

#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;
}