*扩展KMP

今天是2017/7/8,DCDCBigBig的第二十二篇博文

EXKMP(next+extend)

#include
#include
#include
#include
using namespace std;
int l1,l2,next[100001],extend[100001];
char s[100001],t[100001];
void getnext(){
    int a=0,p;
    next[0]=l2;
    for(int i=1,j=-1;iif(j<0||i+next[i-a]>=p){
            if(j<0){
                p=i;
                j=0;
            }
            while(pnext[i]=j;
            a=i;
        }else next[i]=next[i-a];
    }
}
void getex(){
    int a=0,p;
    for(int i=0,j=-1;iif(j<0||i+next[i-a]>=p){
            if(j<0){
                p=i;
                j=0;
            }
            while(ps[p]==t[j]){
                p++;
                j++;
            }
            extend[i]=j;
            a=i;
        }else extend[i]=next[i-a];
    }
}
int main(){
    memset(next,0,sizeof(next));
    scanf("%s%s",s,t);
    l1=strlen(s);
    l2=strlen(t);
    getnext();
    getex();
    for(int i=0;iprintf("%d ",next[i]);
    }
    printf("\n");
    for(int i=0;iprintf("%d ",extend[i]);
    }
    return 0;
}

你可能感兴趣的:(算法-字符串)