KMP算法&拓展KMP算法模板

来源：互联网发布：淘宝隐藏导航代码编辑：程序博客网时间：2024/05/22 06:19

KMP

char a[N],b[M];int nxt[M];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

char a[N],b[M];int nxt[M],ex[N];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;        }    }}

0 0