KMP算法&拓展KMP算法模板

KMP

char a
,b
;
int nxt
;
int n,m;
void Cal_nxt()
{
int i;
nxt[0]=-1;
for(i=1;i<m;i++)
{
int j=nxt[i-1];
while(j>=0&&b[i]!=b[j+1]) j=nxt[j];
nxt[i]=b[i]==b[j+1]?j+1:-1;
}
}
int Kmp()
{
int i,j=0;
for(i=0;i<n&&j<m;i++)
{
while(j>0&&a[i]!=b[j]) j=nxt[j-1]+1;
if(a[i]==b[j]) j++;
}
return (j==m)?(i-m):-1;
}

拓展KMP

const int M=100010;
const int N=100010;
char a
,b[M];
int nxt[M],ex
;
int n,m;
void Cal_nxt()
{
int i=0;
nxt[0]=m;
while(b[0]==b[i+1]&&i+1<m) i++;
nxt[1]=i;
int po=1;
for(i=2;i<m;i++)
{
if(nxt[i-po]+i<nxt[po]+po) nxt[i]=nxt[i-po];
else
{
int j=nxt[po]+po-i;
if(j<0) j=0;
while(i+j<m&&b[j]==b[j+i]) j++;
nxt[i]=j;
po=i;
}
}
}
void Exkmp()
{
int i=0;
cal_nxt();
while(a[i]==b[i]&&i<n&&i<m) i++;
ex[0]=i;
int po=0;
for(i=1;i<n;i++)
{
if(nxt[i-po]+i<ex[po]+po) ex[i]=nxt[i-po];
else
{
int j=ex[po]+po-i;
if(j<0) j=0;
while(i+j<n&&j<m&&a[j+i]==b[j]) j++;
ex[i]=j;
po=i;
}
}
}