ekmp 代码模板

ekmp  自己整理的模板，仅作模板保存使用

#include <bits/stdc++.h>
#define ll long long
using namespace std;

const int MAX=100010; //字符串长度最大值
int net[MAX],extend[MAX];//net数组记录模式串的子串与原串的最长前缀，extend记录模式串与匹配串的子串最长前缀
char ss[MAX],tt[MAX];//ss为匹配串，tt为模式串

void getnet(int len)//求net数组，数组下标从0开始
{
    int i=0,j,po;//po为标记节点
    net[0]=len;
    while(tt[i]==tt[i+1]&&i+1<len) i++;net[1]=i;
    po=1;
    for(i=2;i<len;i++)
    {
        if(net[i-po]+i<net[po]+po) net[i]=net[i-po]; //情况一，直接得结果
        else//情况二，需要向后匹配，并更新po
        {
            j=net[po]+po-i;
            if(j<0) j=0;
            while(i+j<len&&tt[j]==tt[j+i]) j++;
            net[i]=j;po=i;
        }
    }
}

void getextend(int lens,int lent)//求extend数组，lens为匹配串长度，lent为模式串长度
{
    int i=0,j,po;
    while(ss[i]==tt[i]&&i<lens&&i<lent) i++; extend[0]=i;
    po=0;
    for(i=1;i<lens;i++)
    {
        if(net[i-po]+i<extend[po]+po) extend[i]=net[i-po];
        else
        {
            j=extend[po]+po-i;
            if(j<0) j=0;
            while(i+j<lens&&j<lent&&ss[j+i]==tt[j]) j++;
            po=i;extend[i]=j;
        }
    }
}

int main()
{
    cin>>ss;
    cin>>tt;
    int lens=strlen(ss);
    int lent=strlen(tt);
    getnet(lent);
    getextend(lens,lent);
    for(int i=0;i<=lent;i++)
        cout<<net[i]<<" ";
    cout<<endl;
    for(int i=0;i<=lens;i++)
        cout<<extend[i]<<" ";
    cout<<endl;
    return 0;
}