扩展KMP模板

具体原理可以参考这里：扩展KMP、扩展KMP算法（刘毅）

代码：

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
const int N=1e6+5;

int len1,len2;
int nxt[N],extend[N];       //nxt[i]表示t[0...len2-1]和t[i...len2-1]的最大公共前缀长度
                            //extend[i]表示s[i...len1-1]和t[0...len2-1]的最大公共前缀长度

//获得next[i](0<=i<=len(t))，跟ex_kmp()类似，相当于字符串自身t匹配，t[i..len2-1]和t[0...len2-1]和等价于s和t
void getnext(char *t){
    int a,p,len;
    len=strlen(t);
    nxt[0]=len;
    for(int i=1,j=-1;i<len;i++,j--){
        if(j<0||i+nxt[i-a]>=p){
            if(j<0)
                p=i,j=0;
            while(p<len&&t[p]==t[j])
                p++,j++;
            nxt[i]=j;
            a=i;
        }
        else
            nxt[i]=nxt[i-a];
    }
}

//求得extend[i](0=<i<=len(s))
void ex_kmp(char *s,char *t){
    int a,p,len1,len2;               //记录匹配成功的字符的最远位置p，及起始位置a
    len1=strlen(s);
    len2=strlen(t);
    for(int i=0,j=-1;i<len1;i++,j--){//j即等于p与i的距离，其作用是判断i是否大于p（如果j<0，则i大于p）
        if(j<0||i+nxt[i-a]>=p){      //i大于p（其实j最小只可以到-1，j<0的写法方便读者理解程序）
            if(j<0)
                p=i,j=0;
            while(p<len1&&j<len2&&s[p]==t[j])
                p++,j++;
            extend[i]=j;
            a=i;
        }
        else
            extend[i]=nxt[i-a];
    }
}