扩展KMP模板

Posted 2020-06-18 chenhuan001

扩展KMP指的是

对于给出的串S和T，以O(n)的时间求出。

对于所有0<=i<len(S),S(i,i+1,...,len(s)-1)与T的最长前缀长度。

以下是模板：

#include <iostream>
#include <string>
#include <stdio.h>
#include <string.h>
using namespace std;

const int MM=100005;
int mynext[MM],extand[MM];
char S[MM],T[MM];
void Getmynext(const char *T){
    int len=strlen(T),a=0;
    mynext[0]=len;
    while(a<len-1 && T[a]==T[a+1]) a++;
    mynext[1]=a;
    a=1;
    for(int k=2;k<len;k++){
        int p=a+mynext[a]-1,L=mynext[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++;
            mynext[k]=j;
            a=k;
        }
        else
            mynext[k]=L;
    }
}
void GetExtand(const char *S,const char *T){
    Getmynext(T);
    int slen=strlen(S),tlen=strlen(T),a=0;
    int MinLen = slen < tlen ? slen : tlen;
    while(a<MinLen && S[a]==T[a]) a++;
    extand[0]=a;
    a=0;
    for(int k=1;k<slen;k++){
        int p=a+extand[a]-1, L=mynext[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++;
            extand[k]=j;
            a=k;
        }
        else
            extand[k]=L;
    }
}
int main(){
    int tt;
    cin>>tt;
    while(tt--)
    {
        scanf("%s%s",S,T);
        GetExtand(S,T);
        //接下来用mynext 和 extand 即可.
    }
    return 0;
}