扩展KMP模板

Posted 2022-11-26 zjj0624

扩展KMP
扩展KMP是一种解决字符串的数据结构,我们假设母串为S,长度为n,子串为T,长度为m,让我们求出所有S的后缀子串与T的最长公共前缀,也就是求出S(i,n)与T(1,m)的最长公共前缀,定义成一个数组就是$extend[i]$
扩展KMP就是求这个extend[i].
具体extend的求法就不具体写了,这里只写求extend的模板.
模板
用到的数组

char T[N]; //子串
char S[N]; //母串
int next[N];//辅助数组next
int extend[N];//extend数组

求extend数组

void EXTEND_KMP ()

	int lens = strlen (S);
	int lent = strlen (T);
	next[0] = lent;
	int i, j, p, L;
	j = 0;
	while (j + 1 < lent && T[j] == T[j + 1])
	
		++j;
	
	next[1] = j;
	int a = 1;
	for (i = 2; i < lent; ++i)
	
		p = next[a] + a - 1;
		L = next[i - a];
		if (i + L < p + 1)
		
			next[i] = L;
		
		else
		
			j = max(0, p - i + 1);
			while (i + j < lent && T[i + j] == T[j])
			
				++j;
			
			next[i] = j;
			a = i;
		
	
	j = 0;
	while (j < lens && S[j] == T[j])
	
		++j;
	
	extend[0] = j;
	a = 0;
	for (i = 1; i < lens; ++i)
	
		p = extend[a] + a - 1;
		L = next[i - a];
		if (L + i < p + 1)
		
			extend[i] = L;
		
		else
		
			j = max(0, p - i + 1);
			while (i + j < lens && j < lent && S[i + j] == T[j])
			
				++j;
			
			extend[i] = j;
			a = i;