(扩展KMP)模板

来源:互联网 发布:钣金展开软件 编辑:程序博客网 时间:2024/06/05 19:10
const int MAXN=1000005;int  nt[MAXN],ext[MAXN];char S[MAXN],T[MAXN];//母串,匹配串void Getnext(char *T,int *next){    int len=strlen(T),a=0;    next[0]=len;    while(a<len-1 && T[a]==T[a+1]) a++;    next[1]=a;    a=1;    for(int k=2; k<len; k++)    {        int p=a+next[a]-1,L=next[k-a];        if( (k-1)+L >= p)        {            int j = (p-k+1)>0 ? (p-k+1) : 0;            while(k+j<len && T[k+j]==T[j]) j++;            next[k]=j;            a=k;        }        else            next[k]=L;    }}void GetExtend(char *S,char *T,int *next){    Getnext(T,next);    int slen=strlen(S),tlen=strlen(T),a=0;    int MinLen = slen < tlen ? slen : tlen;    while(a<MinLen && S[a]==T[a]) a++;    ext[0]=a;    a=0;    for(int k=1; k<slen; k++)    {        int p=a+ext[a]-1, L=next[k-a];        if( (k-1)+L >= p)        {            int j= (p-k+1) > 0 ? (p-k+1) : 0;            while(k+j<slen && j<tlen && S[k+j]==T[j]) j++;            ext[k]=j;            a=k;        }        else            ext[k]=L;    }}
extend数组的含义:extend[i]存储的是 根据匹配串 存储在母串里面的信息S:babababaT:abaextend:0 3 0 3 0 3 0 1;extend[0] 代表从S的0位置开始,与T公共前缀的长度,为0extend[1] 代表从S的1位置开始,与T公共前缀的长度,为3....extend[7] 代表从S的7位置开始,与T公共前缀的长度,为1我们再来看下extend[1],其值为3,也就是说从s的从1位置开始有三个字符和T相同,那么,T的前缀a在S中的[1,1+3]位置出现了一次,T的前缀ab在S中的[1,1+3]位置出现了一次,T的前缀aba在S中的[1,1+3]位置出现一次,同样extend[3],extend[5] 也是这样,总的来说,T的前缀a出现了四次,T的前缀ab出现了三次,T的前缀aba出现了三次应用:1、求匹配串的前缀在母串出现的次数2、求匹配串的后缀在母串出现的次数,这里需要将匹配串和母串倒置一下,转换成求匹配串的前缀在母串出现的次数3、最长回文子串4、等等吧